Arreglado error Gradio context y agregados decoradores @spaces.GPU
Browse files
app.py
CHANGED
|
@@ -837,6 +837,8 @@ def generate_image(
|
|
| 837 |
error_image = Image.new('RGB', (512, 512), color='red')
|
| 838 |
return error_image
|
| 839 |
|
|
|
|
|
|
|
| 840 |
def generate_video(prompt, model_name, num_frames=16, num_inference_steps=20):
|
| 841 |
"""Generar video con el modelo seleccionado"""
|
| 842 |
try:
|
|
@@ -880,6 +882,8 @@ def generate_video(prompt, model_name, num_frames=16, num_inference_steps=20):
|
|
| 880 |
print(f"Error generando video: {str(e)}")
|
| 881 |
return f"Error generando video: {str(e)}"
|
| 882 |
|
|
|
|
|
|
|
| 883 |
def generate_video_with_info(prompt, model_name, optimization_level="balanced", input_image=None):
|
| 884 |
"""Generar video con informaci贸n adicional - funci贸n para API"""
|
| 885 |
try:
|
|
@@ -1340,18 +1344,18 @@ with gr.Blocks(title="Modelos Libres de IA", theme=gr.themes.Soft()) as demo:
|
|
| 1340 |
outputs=video_output
|
| 1341 |
)
|
| 1342 |
|
| 1343 |
-
# Agregar endpoint para generate_video_with_info
|
| 1344 |
-
demo.load(
|
| 1345 |
-
|
| 1346 |
-
|
| 1347 |
-
|
| 1348 |
-
|
| 1349 |
-
|
| 1350 |
-
|
| 1351 |
-
|
| 1352 |
-
|
| 1353 |
-
|
| 1354 |
-
)
|
| 1355 |
|
| 1356 |
# Configuraci贸n para Hugging Face Spaces
|
| 1357 |
if __name__ == "__main__":
|
|
|
|
| 837 |
error_image = Image.new('RGB', (512, 512), color='red')
|
| 838 |
return error_image
|
| 839 |
|
| 840 |
+
# @spaces.GPU #[uncomment to use ZeroGPU]
|
| 841 |
+
@spaces.GPU
|
| 842 |
def generate_video(prompt, model_name, num_frames=16, num_inference_steps=20):
|
| 843 |
"""Generar video con el modelo seleccionado"""
|
| 844 |
try:
|
|
|
|
| 882 |
print(f"Error generando video: {str(e)}")
|
| 883 |
return f"Error generando video: {str(e)}"
|
| 884 |
|
| 885 |
+
# @spaces.GPU #[uncomment to use ZeroGPU]
|
| 886 |
+
@spaces.GPU
|
| 887 |
def generate_video_with_info(prompt, model_name, optimization_level="balanced", input_image=None):
|
| 888 |
"""Generar video con informaci贸n adicional - funci贸n para API"""
|
| 889 |
try:
|
|
|
|
| 1344 |
outputs=video_output
|
| 1345 |
)
|
| 1346 |
|
| 1347 |
+
# Agregar endpoint para generate_video_with_info dentro del contexto de Gradio
|
| 1348 |
+
demo.load(
|
| 1349 |
+
generate_video_with_info,
|
| 1350 |
+
inputs=[
|
| 1351 |
+
gr.Textbox(label="Prompt", placeholder="Describe el video..."),
|
| 1352 |
+
gr.Dropdown(choices=list(MODELS["video"].keys()), label="Modelo"),
|
| 1353 |
+
gr.Dropdown(choices=["speed", "balanced", "quality"], value="balanced", label="Optimizaci贸n"),
|
| 1354 |
+
gr.Image(label="Imagen de entrada (opcional)", type="pil")
|
| 1355 |
+
],
|
| 1356 |
+
outputs=gr.Video(label="Video Generado", format="mp4"),
|
| 1357 |
+
api_name="generate_video_with_info"
|
| 1358 |
+
)
|
| 1359 |
|
| 1360 |
# Configuraci贸n para Hugging Face Spaces
|
| 1361 |
if __name__ == "__main__":
|