Bartusito commited on
Commit
123f417
1 Parent(s): 906d41e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -19,6 +19,7 @@ def modelo2(text):
19
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
20
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
21
  pipe = pipe.to("cpu")
 
22
  # Conversión a float32
23
  image = pipe(text).images[0]
24
 
 
19
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
20
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
21
  pipe = pipe.to("cpu")
22
+ print(text)
23
  # Conversión a float32
24
  image = pipe(text).images[0]
25