Update app.py

app.py

@@ -74,8 +74,8 @@ def run_gpt(
     for response in stream:
         resp += response.token.text
     
-    history.append((in_prompt,resp))
-    return history
+    #history.append((in_prompt,resp))
+    return resp
 
 
 def run(purpose,history,model_drop):
@@ -89,7 +89,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 (out_prompt,None)
+    history.append((purpose,out_prompt))
+    yield (history,None)
     #out_img = infer(out_prompt)
     model=loaded_model[int(model_drop)]
     out_img=model(out_prompt)
@@ -103,7 +104,7 @@ def run(purpose,history,model_drop):
     if r.status_code == 200:
         out = Image.open(io.BytesIO(r.content))
         #yield ([(purpose,out_prompt)],out)
-        yield (out_prompt,out)
+        yield (history,out)
     else:
         yield ([(purpose,"an Error occured")],None)