Uhhy commited on
Commit
331aa5a
verified
1 Parent(s): f956dd4

Update appxx.py

Browse files
Files changed (1) hide show
  1. appxx.py +4 -1
appxx.py CHANGED
@@ -9,7 +9,10 @@ import spaces
9
  def initialize_pipeline():
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  model_id = "CompVis/stable-diffusion-v1-4" # Cambia esto al ID del modelo adecuado
12
- pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id).to(device)
 
 
 
13
  return pipe
14
 
15
  # Configuraci贸n inicial del modelo
 
9
  def initialize_pipeline():
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  model_id = "CompVis/stable-diffusion-v1-4" # Cambia esto al ID del modelo adecuado
12
+ try:
13
+ pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id).to(device)
14
+ except ValueError as e:
15
+ return f"Error cargando el pipeline: {e}"
16
  return pipe
17
 
18
  # Configuraci贸n inicial del modelo