Update app.py

app.py

@@ -110,49 +110,49 @@ def generate(
     if not use_negative_prompt_2:
         negative_prompt_2 = None  # type: ignore
 
-if use_inpainting:
-    images = pipe(
-        prompt=prompt,
-        image=img_url,
-        mask_image=mask_url,
-        strength=strength_img2img,
-        negative_prompt=negative_prompt,
-        prompt_2=prompt_2,
-        negative_prompt_2=negative_prompt_2,
-        guidance_scale=guidance_scale_base,
-        num_inference_steps=num_inference_steps_base,
-        generator=generator,
-    ).images[0]
-    return images
-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]
+    if use_inpainting:
+        images = pipe(
+            prompt=prompt,
+            image=img_url,
+            mask_image=mask_url,
+            strength=strength_img2img,
+            negative_prompt=negative_prompt,
+            prompt_2=prompt_2,
+            negative_prompt_2=negative_prompt_2,
+            guidance_scale=guidance_scale_base,
+            num_inference_steps=num_inference_steps_base,
+            generator=generator,
+        ).images[0]
+        return images
+    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(