Update app.py

app.py

@@ -138,35 +138,6 @@ 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(