Update app.py

app.py

@@ -85,12 +85,9 @@ def generate(
             init_image = Image.open(BytesIO(response.content)).convert("RGB")
             init_image = init_image.resize((width, height))
 
-            response = requests.get(img_url)
-            image_init = Image.open(BytesIO(response.content)).convert("RGB")
-
-            response = requests.get(mask_url)
-            mask_image = Image.open(BytesIO(response.content)).convert("RGB")
-
+            image_init = load_image(img_url)
+            mask_image = load_image(mask_url)
+            
         if use_lora:
             pipe.load_lora_weights(lora)
             pipe.fuse_lora(lora_scale)
@@ -127,6 +124,35 @@ def generate(
             generator=generator,
         ).images[0]
         return image
+    elif use_img2img:
+        images = pipe(
+            prompt=prompt,
+            image=init_image,
+            strength=strength_img2img,
+            negative_prompt=negative_prompt,
+            prompt_2=prompt_2,
+            negative_prompt_2=negative_prompt_2,
+            width=width,
+            height=height,
+            guidance_scale=guidance_scale_base,
+            num_inference_steps=num_inference_steps_base,
+            generator=generator,
+            output_type="pil",
+        ).images[0]
+        return images
+    else:
+        return pipe(
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            prompt_2=prompt_2,
+            negative_prompt_2=negative_prompt_2,
+            width=width,
+            height=height,
+            guidance_scale=guidance_scale_base,
+            num_inference_steps=num_inference_steps_base,
+            generator=generator,
+            output_type="pil",
+        ).images[0]
 
 with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
     gr.HTML(