A

app.py

@@ -70,6 +70,7 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto" if device == "cuda" else None
 )
 
+@spaces.GPU(duration=120)
 def construir_prompt(input_text):
     return f"""
 {instrucciones}
@@ -80,6 +81,7 @@ Texto original a mejorar:
 Versión mejorada y finalizada de la resolución judicial:
 """
 
+@spaces.GPU(duration=120)
 def post_procesar_resolucion(texto):
     secciones = ['VISTOS:', 'CONSIDERANDO:', 'SE RESUELVE:']
     for seccion in secciones:
@@ -91,6 +93,7 @@ def post_procesar_resolucion(texto):
     
     return texto
 
+@spaces.GPU(duration=120)
 def validar_resolucion(texto):
     requisitos = [
         ('VISTOS:', "La sección 'VISTOS:' es obligatoria"),
@@ -107,6 +110,7 @@ def validar_resolucion(texto):
     
     return errores
 
+@spaces.GPU(duration=120)
 def mejorar_resolucion(input_text):
     prompt = construir_prompt(input_text)
     inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True, max_length=1024).to(device)