Update app.py

app.py

@@ -48,9 +48,6 @@ def pipe_i2i():
     PIPE.unet = torch.compile(PIPE.unet, mode="reduce-overhead", fullgraph=True)
     return PIPE
 
-pp1=pipe_t2i()
-pp2=pipe_i2i()
-
 def translate(text,lang):
     if text == None or lang == None:
         return ""       
@@ -131,18 +128,6 @@ def tok(txt):
     print(toks)
     return toks
 
-def main(p1,p2,*result):
-    p1_en = translate(p1,"english")
-    p2_en = translate(p2,"english")
-    p = {"a":p1_en,"b":p2_en}
-    ln = len(result)
-    rng = list(range(ln))
-    arr = [p for _ in rng]
-    out = None
-    with Pool(ln, initializer=init_pool, initargs=(pp1,pp2)) as pool:
-        out = pool.imap(infer,arr)
-    return list(out)
-
 def infer(p):
     p1 = p["a"]
     p2 = p["b"]
@@ -209,8 +194,27 @@ function custom(){
 }
 """
 
-def ui():
+if __name__ == "__main__":
+    
+    pp1=pipe_t2i()
+    pp2=pipe_i2i()
+    
     result = []
+    
+    def main(p1,p2,*result):
+        p1_en = translate(p1,"english")
+        p2_en = translate(p2,"english")
+        p = {"a":p1_en,"b":p2_en}
+        ln = len(result)
+        rng = list(range(ln))
+        arr = [p for _ in rng]
+        out = None
+        with Pool(ln, initializer=init_pool, initargs=(pp1,pp2)) as pool:
+            out = pool.imap(infer,arr)
+        return list(out)
+    port_inc()
+    mp.set_start_method("spawn", force=True)
+
     with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
         with gr.Column(elem_id="col-container"):
             gr.Markdown(f"""
@@ -237,9 +241,4 @@ def ui():
                 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=main,inputs=[prompt,prompt2,*result],outputs=result)
-        demo.queue().launch(server_port=int(os.getenv("CUSTOM_PORT")))
-
-if __name__ == "__main__":
-    port_inc()
-    mp.set_start_method("spawn", force=True)
-    ui()
+        demo.queue().launch(server_port=int(os.getenv("CUSTOM_PORT")))