Update app.py

app.py

@@ -193,13 +193,17 @@ def run(p1,p2,*result):
         out1 = None
         print("Starting pool!")
         with Pool(ln) as pool:
-            out1 = pool.imap(infer1,arr1)
-        lst = list(out1.get())
+            out1 = pool.map(infer1,arr1)
+            pool.join()
+            pool.close()
+        lst = list(out1)
         arr2 = [{"a":p1_en,"b":p2_en,"c":lst[_]} for _ in rng]
         out2 = None
         with Pool(ln) as pool:
-            out2 = pool.imap(infer2,arr2)
-        return list(out2.get())
+            out2 = pool.map(infer2,arr2)
+            pool.join()
+            pool.close()
+        return list(out2)
 
 def ui():
     with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo: