Update app.py

app.py

@@ -160,14 +160,7 @@ function custom(){
 }
 """
 
-def main():
-
-    pp1=pipe_t2i()
-    pp2=pipe_i2i()
-    
-    result = []
-
-    def infer(p):
+def infer(p):
         p1 = p["a"]
         p2 = p["b"]
         name = generate_random_string(12)+".png"
@@ -193,8 +186,8 @@ def main():
         else:
             output2.images[0].save("_"+name)
         return "_"+name
-    
-    def run(p1,p2,*result):
+
+def run(p1,p2,*result):
         p1_en = translate(p1,"english")
         p2_en = translate(p2,"english")
         p = {"a":p1_en,"b":p2_en}
@@ -205,10 +198,8 @@ def main():
         with Pool(ln, initializer=init_pool) as pool:
             out = pool.imap(infer,arr)
         return list(out)
-    port_inc()
-    
-    mp.set_start_method("fork", force=True)
 
+def ui():
     with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         with gr.Column(elem_id="col-container"):
             gr.Markdown(f"""
@@ -238,4 +229,16 @@ def main():
         demo.queue().launch(server_port=int(os.getenv("CUSTOM_PORT")))
 
 if __name__ == "__main__":
-    main()
+    
+    global pp1
+    global pp2
+    global result
+    
+    pp1=pipe_t2i()
+    pp2=pipe_i2i()
+    result=[]
+
+    mp.set_start_method("spawn", force=True)
+    
+    port_inc()
+    ui()