Update app.py

app.py

@@ -21,17 +21,16 @@ from diffusers.utils import load_image
 #import jax
 #import jax.numpy as jnp
 
-os.environ["GRADIO_SERVER_PORT"]="7860"
-
 def port_inc():
-    os.environ["GRADIO_SERVER_PORT"]=str(int(os.getenv("GRADIO_SERVER_PORT"))+1)
+    env = os.getenv("CUSTOM_PORT")
+    if env == None:
+        os.environ["CUSTOM_PORT"]="8000"
+    else:
+        os.environ["CUSTOM_PORT"]=str(int(env)+1)
 
 def init_pool():
     port_inc()
 
-_dynamo.config.suppress_errors = True
-mp.set_start_method("spawn", force=True)
-
 #pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to(device)
 #pipe2 = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to(device)
 #pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
@@ -209,32 +208,38 @@ function custom(){
 }
 """
 
-with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
+if __name__ == "__main__":
+    
+    port_inc()
+    _dynamo.config.suppress_errors = True
+    mp.set_start_method("spawn", force=True)
     result = []
-    with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""
-            # MULTI-LANGUAGE IMAGE GENERATOR
-        """)
-        with gr.Row():
-            prompt = gr.Textbox(
-                elem_id="prompt",
-                placeholder="INCLUDE",
-                container=False,
-                max_lines=1
-            )
-        with gr.Row():
-            prompt2 = gr.Textbox(
-                elem_id="prompt2",
-                placeholder="EXCLUDE",
-                container=False,
-                max_lines=1
-            )
-        with gr.Row():
-            run_button = gr.Button("START",elem_classes="btn",scale=0)
-        with gr.Row():
-            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))
-            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],outputs=result)
-    demo.queue().launch()
+    
+    with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
+        with gr.Column(elem_id="col-container"):
+            gr.Markdown(f"""
+                # MULTI-LANGUAGE IMAGE GENERATOR
+            """)
+            with gr.Row():
+                prompt = gr.Textbox(
+                    elem_id="prompt",
+                    placeholder="INCLUDE",
+                    container=False,
+                    max_lines=1
+                )
+            with gr.Row():
+                prompt2 = gr.Textbox(
+                    elem_id="prompt2",
+                    placeholder="EXCLUDE",
+                    container=False,
+                    max_lines=1
+                )
+            with gr.Row():
+                run_button = gr.Button("START",elem_classes="btn",scale=0)
+            with gr.Row():
+                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))
+                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],outputs=result)
+        demo.queue().launch(server_port=os.getenv("CUSTOM_PORT"))