Update app.py

app.py

@@ -151,8 +151,8 @@ function custom(){
 }
 """
 
-def infer(p):
-        print("infer: started")
+def infer1(p):
+        print("infer1: started")
         p1 = p["a"]
         print(f'prompt 1: {p1}')
         p2 = p["b"]
@@ -162,20 +162,23 @@ def infer(p):
         _do = ['beautiful', 'playful', 'photographed', 'realistic', 'dynamic poze', 'deep field', 'reasonable coloring', 'rough texture', 'best quality', 'focused']
         if p1 != "":
             _do.append(f'{p1}')
-        if p2 != "":
-            _dont = f'{p2} where in {p1}'
-            neg = _dont
-        else:
-            neg = None
         posi = " ".join(_do)
         print(posi)
         output = Piper1(posi)
-        print(output)
         output.images[0].save(name)
-        if neg == None:
+        return name
+def infer2(p):
+        print("infer2: started")
+        p1 = p["a"]
+        p2 = p["b"]
+        name = p["c"]
+        if p2 != "":
+            _dont = f'{p2} where in {p1}'
+            neg = _dont
+        else:
             return name
         img = load_image(name).convert("RGB")
-        output2 = Piper2(img,posi,neg)
+        output2 = Piper2(img,p1,neg)
         output2.images[0].save("_"+name)
         return "_"+name
 
@@ -186,12 +189,17 @@ def run(p1,p2,*result):
         ln = len(result)
         print("images: "+str(ln))
         rng = list(range(ln))
-        arr = [p for _ in rng]
-        out = None
+        arr1 = [p for _ in rng]
+        out1 = None
         print("Starting pool!")
         with Pool(ln) as pool:
-            out = pool.map(infer,arr)
-        return list(out.get())
+            out1 = pool.map(infer1,arr1)
+        lst = list(out1.get())
+        arr2 = [{"a":p1_en,"b":p2_en,"c":lst[_]} for _ in rng]
+        out2 = None
+        with Pool(ln) as pool:
+            out2 = pool.map(infer1,arr2)
+        return list(out2.get())
 
 def ui():
     with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
@@ -232,6 +240,6 @@ if __name__ == "__main__":
     pp1=pipe_t2i()
     pp2=pipe_i2i()
 
-    mp.set_start_method("spawn", force=True)
+    mp.set_start_method("fork", force=True)
     #port_inc()
     ui()