Update interface.py

interface.py

@@ -40,7 +40,7 @@ def parse_bounds(bounds_str, num_params):
         upper_bounds = [np.inf] * num_params
         return lower_bounds, upper_bounds
 
-def generate_analysis(prompt, max_length=1024):
+def generate_analysis(prompt, max_length=100):  # Reducido a 100
     """
     Genera un análisis utilizando el modelo Yi-Coder-9B-Chat.
     """
@@ -51,7 +51,7 @@ def generate_analysis(prompt, max_length=1024):
         with torch.no_grad():
             outputs = model.generate(
                 **inputs,
-                max_new_tokens=max_length,
+                max_new_tokens=max_length,  # Limitar la generación a 100 tokens
                 eos_token_id=tokenizer.eos_token_id,
                 pad_token_id=tokenizer.eos_token_id
             )
@@ -62,7 +62,7 @@ def generate_analysis(prompt, max_length=1024):
         print(f"Error al generar el análisis con Yi-Coder: {e}. Usando análisis por defecto.")
         return "Análisis generado por el modelo de lenguaje."
 
-@gpu_decorator(duration=600)  # Ajusta la duración según tus necesidades
+@gpu_decorator(duration=100)  # Reducido de 600 a 100 segundos
 def process_and_plot(
     file,
     biomass_eq1, biomass_eq2, biomass_eq3,
@@ -141,7 +141,6 @@ def process_and_plot(
                 bounds=(lower_bounds, upper_bounds)
             )
             biomass_results.append({
-                # 'model': main_model,  # Eliminado para evitar problemas de serialización
                 'y_pred': y_pred.tolist(),  # Convertir a lista para serialización
                 'equation': equation,
                 'params': main_model.params['biomass']
@@ -169,7 +168,6 @@ def process_and_plot(
                 bounds=(lower_bounds, upper_bounds)
             )
             substrate_results.append({
-                # 'model': main_model,  # Eliminado para evitar problemas de serialización
                 'y_pred': y_pred.tolist(),  # Convertir a lista para serialización
                 'equation': equation,
                 'params': main_model.params['substrate']
@@ -197,7 +195,6 @@ def process_and_plot(
                 bounds=(lower_bounds, upper_bounds)
             )
             product_results.append({
-                # 'model': main_model,  # Eliminado para evitar problemas de serialización
                 'y_pred': y_pred.tolist(),  # Convertir a lista para serialización
                 'equation': equation,
                 'params': main_model.params['product']
@@ -257,6 +254,6 @@ Analiza los siguientes resultados experimentales y proporciona un veredicto sobr
 """
 
     # Generar el análisis utilizando Yi-Coder
-    analysis = generate_analysis(prompt, max_length=1024)
+    analysis = generate_analysis(prompt, max_length=100)  # Reducido a 100
 
     return image, analysis