Update app.py

app.py

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