Update app.py

app.py

@@ -95,7 +95,7 @@ def generate_random_string(length):
     return ''.join(random.choice(characters) for _ in range(length))
 
 @spaces.GPU(duration=35)
-def Piper(_do):
+def Piper(pipe,_do):
     pipe = pipe_t2i()
     try:
         retu = pipe(
@@ -112,8 +112,7 @@ def Piper(_do):
         return None
 
 @spaces.GPU(duration=35)
-def Piper2(img,posi,neg):
-    pipe = pipe_i2i()
+def Piper2(pipe,img,posi,neg):
     try:
         retu = pipe2(
             prompt=posi,
@@ -131,6 +130,19 @@ def tok(txt):
     print(toks)
     return toks
 
+def main(p1,p2):
+    p1_en = translate(p1,"english")
+    p2_en = translate(p2,"english")
+    p = {"a":p1_en,"b":p2_en}
+    ln = len(result)
+    rng = range(ln)
+    for _ in rng:
+        b.put(p)
+    with mp.Pool(ln, initializer=init_pool, initargs=(b,)) as pool:
+        out = pool.map(infer)
+        pool.clear()
+        return list(out.get())
+
 def infer():
     p1 = b["a"]
     p2 = b["b"]
@@ -143,16 +155,15 @@ def infer():
         neg = _dont
     else:
         neg = None
-    output = Piper('A '+" ".join(_do))
+    output = Piper(pipe_t2i(),'A '+" ".join(_do))
     if output == None:
         return None
     else:
         output.images[0].save(name)
     if neg == None:
         return name
-
     img = load_image(name).convert("RGB")
-    output2 = Piper2(img,p1,neg)
+    output2 = Piper2(pipe_i2i(),img,p1,neg)
     if output2 == None:
         return None
     else:
@@ -225,24 +236,5 @@ 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 main(p1,p2):
-        
-        p1_en = translate(p1,"english")
-        p2_en = translate(p2,"english")
-        
-        p = {"a":p1_en,"b":p2_en}
-        ln = len(result)
-        rng = range(ln)
-        
-        for _ in rng:
-            b.put(p)
-
-        with mp.Pool(ln, initializer=init_pool, initargs=(b,)) as pool:
-            out = pool.map(infer)
-            pool.clear()
-            return list(out.get())
-        
-        #return list( _ret(p1_en,p2_en) )
-        
     run_button.click(fn=main,inputs=[prompt,prompt2],outputs=result)
     demo.queue().launch()