Update app.py

app.py

@@ -166,6 +166,7 @@ def infer1(p):
         output = Piper1(posi)
         output.images[0].save(name)
         return name
+
 def infer2(p):
         print("infer2: started")
         p1 = p["a"]
@@ -181,7 +182,7 @@ def infer2(p):
         output2.images[0].save("_"+name)
         return "_"+name
 
-def run(p1,p2,*result):
+def run1(p1,p2,*result):
         p1_en = translate(p1,"english")
         p2_en = translate(p2,"english")
         p = {"a":p1_en,"b":p2_en}
@@ -195,6 +196,16 @@ def run(p1,p2,*result):
         pool1.close()
         pool1.join()
         pool1.clear()
+
+        return out1
+
+def run2(p1,p2,*result):
+        p1_en = translate(p1,"english")
+        p2_en = translate(p2,"english")
+        p = {"a":p1_en,"b":p2_en}
+        ln = len(result)
+        print("images: "+str(ln))
+        rng = list(range(ln))
     
         arr2 = [{"a":p1_en,"b":p2_en,"c":out1[_]} for _ in rng]
         pool2 = Pool(ln)
@@ -242,7 +253,7 @@ def main():
                 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))
 
-        run_button.click(fn=run,inputs=[prompt,prompt2,*result],outputs=result)
+        run_button.click(fn=run1,inputs=[prompt,prompt2,*result],outputs=result).then(fn=run2,inputs=[prompt,prompt2,*result],outputs=result)
         demo.queue().launch()
 
 if __name__ == "__main__":