Update app.py

app.py

@@ -1,6 +1,6 @@
 import torch
 from controlnet_aux import CannyDetector
-from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
+from diffusers import FluxControlPipeline
 from diffusers.utils import load_image
 import gradio as gr
 import random
@@ -17,12 +17,9 @@ os.system("pip install -U diffusers controlnet_aux mediapipe")
 MAX_SEED = 2**32 - 1
 MAX_IMAGE_SIZE = 1024
 
-# Inicialización del modelo ControlNet
-controlnet = ControlNetModel.from_pretrained(
+# Inicialización del pipeline Flux Canny Dev
+pipe = FluxControlPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-Canny-dev", torch_dtype=torch.float16
-)
-pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "CompVis/stable-diffusion-v1-4", controlnet=controlnet, torch_dtype=torch.float16
 ).to("cuda" if torch.cuda.is_available() else "cpu")
 
 # Procesador Canny