Update app.py

app.py

@@ -72,9 +72,10 @@ def run(purpose,history,model_drop,chat_drop,choice):
     if choice == "Generate":
         #out_img = infer(out_prompt)
         out_prompt=generate(purpose,history,model_drop,chat_drop)
+        history.append((purpose,out_prompt))
+        yield (history,None)        
         model=loaded_model[int(model_drop)]
         out_img=model(out_prompt)
-        history.append((purpose,out_prompt))
         #return (history,None)
         print(out_img)
         url=f'https://johann22-chat-diffusion-describe.hf.space/file={out_img}'
@@ -86,13 +87,14 @@ def run(purpose,history,model_drop,chat_drop,choice):
         if r.status_code == 200:
             out = Image.open(io.BytesIO(r.content))
             #yield ([(purpose,out_prompt)],out)
-            return (history,out)
+            yield (history,out)
         else:
-            return ([(purpose,"an Error occured")],None)
+            yield ([(purpose,"an Error occured")],None)
     if choice == "Describe":
         #out_img = infer(out_prompt)
         out_prompt=describe(purpose,history,model_drop,chat_drop)
         history.append((purpose,out_prompt))
+        yield (history,None)        
         model=loaded_model[int(model_drop)]
         out_img=model(out_prompt)
         print(out_img)
@@ -105,9 +107,9 @@ def run(purpose,history,model_drop,chat_drop,choice):
         if r.status_code == 200:
             out = Image.open(io.BytesIO(r.content))
             #yield ([(purpose,out_prompt)],out)
-            return (history,out)
+            yield (history,out)
         else:
-            return ([(purpose,"an Error occured")],None)        
+            yield ([(purpose,"an Error occured")],None)        
 ################################################
 style="""
 .top_head{