Update app.py

app.py

@@ -197,8 +197,10 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         return v
     def _rets(p1,p2):
         ln = len(result)
-        idxs = [ [idx,p1,p2] for idx in list(range(ln)) ]
-        return list(Pool(ln).imap( _ret, idxs ))
+        idxs = list(range(ln))
+        p1s = [p1 for _ in idxs]
+        p2s = [p2 for _ in idxs]
+        return list(Pool(ln).imap( _ret, idxs, p1s, p2s ))
     run_button.click(fn=_rets,inputs=[prompt,prompt2],outputs=result)
 
 demo.queue().launch()