Update app.py

app.py

@@ -67,28 +67,6 @@ def generate(
     mask_url = '',
 ):
     if torch.cuda.is_available():
-            
-        if not use_img2img:
-            pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
-
-            if use_vae:  
-                vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
-                pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
-            
-        if use_img2img:
-            pipe = AutoPipelineForImage2Image.from_pretrained(model, torch_dtype=torch.float16)
-
-            if use_vae:  
-                vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
-                pipe = AutoPipelineForImage2Image.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
-                
-        if use_inpainting:
-            pipe = AutoPipelineForInpainting.from_pretrained(model, torch_dtype=torch.float16)
-
-            image_init = load_image(img_url)
-            mask_image = load_image(mask_url)
-
-        if use_controlnet:
             controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
             pipe = StableDiffusionControlNetPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16)
 
@@ -131,6 +109,7 @@ def generate(
             image=image,
             control_image=canny_image,
         ).images[0]
+        return image
 
 with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
     gr.HTML(