Update interface.py

interface.py

@@ -11,6 +11,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 from sympy import symbols, sympify, lambdify
 import copy
 from config import DEVICE, MODEL_PATH, MAX_LENGTH, TEMPERATURE
+from decorators import spaces  # Importamos el decorador
 
 # Configuración del dispositivo
 device = DEVICE
@@ -22,7 +23,7 @@ model = AutoModelForCausalLM.from_pretrained(model_path)
 model.to(device)
 model.eval()
 
-@torch.no_grad()
+@spaces.GPU(duration=300)  # Aplicamos el decorador para asegurar uso de GPU
 def generate_analysis(prompt, max_length=MAX_LENGTH):
     try:
         input_ids = tokenizer.encode(prompt, return_tensors='pt').to(device)