Hugging Face's logo

Spaces:
iespsurmreqmer
/
DiffusersTransformers
No application file

App Files Community
iespsurmreqmer commited on
Commit
ad8cb4d
·
1 Parent(s): 9eb2dbb

.

Browse files
Files changed (1) hide show
  1. app3.py +1 -1
app3.py CHANGED
@@ -5,7 +5,7 @@ from PIL import Image
5
  # Cargar el pipeline
6
  model_id = "stabilityai/stable-diffusion-x4-upscaler"
7
  pipe = StableDiffusionUpscalePipeline.from_pretrained(model_id, torch_dtype=torch.float16)
8
- pipe = pipe.to("cuda") # usa "cpu" si no tenés GPU
9
 
10
  # Cargar la imagen a mejorar
11
  input_image = Image.open("mi_imagen.jpg").convert("RGB")
 
5
  # Cargar el pipeline
6
  model_id = "stabilityai/stable-diffusion-x4-upscaler"
7
  pipe = StableDiffusionUpscalePipeline.from_pretrained(model_id, torch_dtype=torch.float16)
8
+ pipe = pipe.to("cpu")
9
 
10
  # Cargar la imagen a mejorar
11
  input_image = Image.open("mi_imagen.jpg").convert("RGB")