eloi commited on
Commit
f08ce27
1 Parent(s): 6e5d0ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ pipe = pipe.to("cuda") # Asegúrate de tener una GPU disponible o cambia esto a
11
 
12
  def inferencia(prompt):
13
  # Realizar la inferencia
14
- imagen = pipe(prompt, num_inference_steps=50, guidance_scale=7, height=720, width=512).images[0] # Modifica num_inference_steps según sea necesario
15
 
16
  # Convertir la imagen a un formato que Gradio pueda utilizar, si es necesario
17
  # Esto podría incluir convertir un tensor de PyTorch a una imagen PIL y luego a una matriz NumPy, por ejemplo
 
11
 
12
  def inferencia(prompt):
13
  # Realizar la inferencia
14
+ imagen = pipe(prompt, num_inference_steps=64, guidance_scale=7, height=720, width=512).images[0] # Modifica num_inference_steps según sea necesario
15
 
16
  # Convertir la imagen a un formato que Gradio pueda utilizar, si es necesario
17
  # Esto podría incluir convertir un tensor de PyTorch a una imagen PIL y luego a una matriz NumPy, por ejemplo