Update app.py

app.py

@@ -73,15 +73,14 @@ def run_gpt(
     resp = ""
     for response in stream:
         resp += response.token.text
-    return resp
+    
+    history.append((in_prompt,resp))
+    return history
 
 
 def run(purpose,history,model_drop):
     print (history)
-    #print(purpose)
-    #print(hist)
     task=None
-    directory="./"
     #if history:
     #    history=str(history).strip("[]")
     #if not history:
@@ -89,7 +88,8 @@ def run(purpose,history,model_drop):
 
     #action_name, action_input = parse_action(line)
     out_prompt = run_gpt(purpose,history)
-    yield ([(purpose,out_prompt)],None)
+    #yield ([(purpose,out_prompt)],None)
+    yield (out_prompt,None)
     #out_img = infer(out_prompt)
     model=loaded_model[int(model_drop)]
     out_img=model(out_prompt)
@@ -102,7 +102,8 @@ def run(purpose,history,model_drop):
     r = requests.get(url, stream=True)
     if r.status_code == 200:
         out = Image.open(io.BytesIO(r.content))
-        yield ([(purpose,out_prompt)],out)
+        #yield ([(purpose,out_prompt)],out)
+        yield (out_prompt,out)
     else:
         yield ([(purpose,"an Error occured")],None)