Spaces:

Ntdeseb
/

ntia

Running

Ntdeseb commited on Jul 31

Commit

df5c53d

1 Parent(s): fa6f4ac

Restaurar @spaces.GPU solo en generate_image para habilitar GPU

Files changed (1) hide show

app.py CHANGED Viewed

@@ -731,7 +731,7 @@ def generate_text(prompt, model_name, max_length=100):
     except Exception as e:
         return f"Error generando texto: {str(e)}"
-# @spaces.GPU
 def generate_image(prompt, model_name, negative_prompt="", seed=0, width=1024, height=1024, guidance_scale=7.5, num_inference_steps=20):
     """Generar imagen optimizada para H200"""
     try:

     except Exception as e:
         return f"Error generando texto: {str(e)}"
+@spaces.GPU  # Requerido para habilitar GPU en Hugging Face Spaces
 def generate_image(prompt, model_name, negative_prompt="", seed=0, width=1024, height=1024, guidance_scale=7.5, num_inference_steps=20):
     """Generar imagen optimizada para H200"""
     try: