Update app.py

app.py

@@ -131,7 +131,7 @@ def tok(txt):
     print(toks)
     return toks
 
-def main(p1,p2,result):
+def main(p1,p2,*result):
     p1_en = translate(p1,"english")
     p2_en = translate(p2,"english")
     p = {"a":p1_en,"b":p2_en}
@@ -236,7 +236,7 @@ def ui():
                 result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
                 result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
-        run_button.click(fn=main,inputs=[prompt,prompt2,result],outputs=result)
+        run_button.click(fn=main,inputs=[prompt,prompt2,*result],outputs=result)
         demo.queue().launch(server_port=int(os.getenv("CUSTOM_PORT")))
 
 if __name__ == "__main__":