Update app.py

app.py

@@ -23,6 +23,7 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
 ENABLE_USE_LORA = os.getenv("ENABLE_USE_LORA", "1") == "1"
 ENABLE_USE_VAE = os.getenv("ENABLE_USE_VAE", "1") == "1"
+ENABLE_USE_IMG2IMG = os.getenv("ENABLE_USE_VAE", "1") == "1"
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
@@ -113,7 +114,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
         model = gr.Text(label='Model', placeholder='e.g. stabilityai/stable-diffusion-xl-base-1.0')
         vaecall = gr.Text(label='VAE', placeholder='e.g. madebyollin/sdxl-vae-fp16-fix')
         lora = gr.Text(label='LoRA', placeholder='e.g. nerijs/pixel-art-xl')
-        url = gr.Text(label='URL', placeholder='URL for image')
+        url = gr.Text(label='URL', placeholder='URL')
         lora_scale = gr.Slider(
                 info="The closer to 1, the more it will resemble LoRA, but errors may be visible.",
                 label="Lora Scale",
@@ -122,7 +123,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
                 step=0.01,
                 value=0.7,
             )
-        use_url = gr.Image(label='Img2Img')
         with gr.Row():
             prompt = gr.Text(
                 placeholder="Input prompt",
@@ -135,6 +135,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
         result = gr.Image(label="Result", show_label=False)
     with gr.Accordion("Advanced options", open=False):
         with gr.Row():
+            use_img2img = gr.Checkbox(label='Use Img2Img', value=False, visible=ENABLE_USE_IMG2IMG)
             use_vae = gr.Checkbox(label='Use VAE', value=False, visible=ENABLE_USE_VAE)
             use_lora = gr.Checkbox(label='Use Lora', value=False, visible=ENABLE_USE_LORA)
             use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=False)
@@ -245,9 +246,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
         queue=False,
         api_name=False,
     )
-    use_url.change(
+    use_img2img.change(
         fn=lambda x: gr.update(visible=x),
-        inputs=use_url,
+        inputs=use_img2img,
         outputs=url,
         queue=False,
         api_name=False,