.

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
-from diffusers import StableDiffusionUpscalePipeline
+from diffusers import DiffusionPipeline
 from PIL import Image
+import torch
 
 # Cargar pipeline (float32 para CPU)
 model_id = "stabilityai/stable-diffusion-x4-upscaler"
-pipe = StableDiffusionUpscalePipeline.from_pretrained(model_id)
-pipe = pipe.to("cpu")  # o "cuda" si hay GPU
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler", dtype=torch.bfloat16, device_map="cuda")
 
 # Función de Gradio
 def upscaler_gradio(imagen):