Update interface.py

interface.py

@@ -12,14 +12,14 @@ 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  # Import the spaces class
 
 device = DEVICE
 model_path = MODEL_PATH
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path).to(device).eval()
 
-from decorators import spaces
-@spaces.GPU(duration=100)
+@spaces.GPU(duration=100)  # Apply the GPU decorator
 def generate_analysis(prompt, max_length=MAX_LENGTH):
     try:
         input_ids = tokenizer.encode(prompt, return_tensors='pt').to(device)
@@ -50,6 +50,7 @@ def parse_bounds(bounds_str, num_params):
         upper_bounds = [np.inf] * num_params
         return lower_bounds, upper_bounds
 
+@spaces.GPU(duration=100)
 def process_and_plot(
     file,
     biomass_eq1, biomass_eq2, biomass_eq3,