Update app.py

app.py

@@ -8,7 +8,7 @@ import requests
 import gradio as gr
 import numpy as np
 from lxml.html import fromstring
-from pathos.multiprocessing import ProcessPool as Pool
+#from pathos.multiprocessing import ProcessPool as Pool
 #from pathos.threading import ThreadPool as Pool
 #from diffusers.pipelines.flux import FluxPipeline
 #from diffusers.utils import export_to_gif
@@ -217,19 +217,27 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
             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))
 
-    def _ret(idx,p1,p2):
-        print(f'Starting {idx}')
+    def _ret(p1,p2):
+        
+        print(f'Starting!')
         v = infer(p1,p2)
-        print(f'Finished {idx}')
+        print(f'Finished!')
         return v
+        
     def _rets(p1,p2):
+        
         p1_en = translate(p1,"english")
         p2_en = translate(p2,"english")
-        ln = len(result)
-        idxs = list(range(ln))
-        p1s = [p1_en for _ in idxs]
-        p2s = [p2_en for _ in idxs]
-        return list(Pool(ln).imap( _ret, idxs, p1s, p2s ))
+
+        #ln = len(result)
+        #idxs = list(range(ln))
+        #p1s = [p1_en for _ in idxs]
+        #p2s = [p2_en for _ in idxs]
+        #pool = Pool(ln)
+        #return list( pool.imap( _ret, p1s, p2s ) )
+
+        return list( _ret(p1_en,p2_en) )
+        
     run_button.click(fn=_rets,inputs=[prompt,prompt2],outputs=result)
 
 demo.queue().launch()