Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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
|