Update app.py

app.py

@@ -167,7 +167,7 @@ with gr.Blocks() as app:
     json_out=gr.JSON()
         
     sub_b = submit_b.click(generate, [msg,chatbot],[chatbot,json_out])
-    sub_im = submit_im.click(run, [json_out,model_drop],[chatbot,im_out])
+    sub_im = submit_im.click(run, [json_out,model_drop],[im_out])
     sub_e = msg.submit(run, [msg, chatbot,model_drop], [chatbot,im_out])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e])
 app.launch()