Update interface.py
Browse files- interface.py +14 -45
interface.py
CHANGED
|
@@ -5,8 +5,6 @@ import pandas as pd
|
|
| 5 |
import matplotlib.pyplot as plt
|
| 6 |
from PIL import Image
|
| 7 |
import io
|
| 8 |
-
from transformers import pipeline
|
| 9 |
-
import torch
|
| 10 |
|
| 11 |
from bioprocess_model import BioprocessModel
|
| 12 |
from decorators import gpu_decorator # Asegúrate de que la ruta es correcta
|
|
@@ -30,31 +28,11 @@ def parse_bounds(bounds_str, num_params):
|
|
| 30 |
|
| 31 |
@gpu_decorator(duration=300)
|
| 32 |
def generate_analysis(prompt, max_length=1024, device=None):
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
:param device: Dispositivo a utilizar ('cpu' o 'cuda').
|
| 39 |
-
:return: Texto generado por el modelo de lenguaje.
|
| 40 |
-
"""
|
| 41 |
-
try:
|
| 42 |
-
# Seleccionar el dispositivo
|
| 43 |
-
if device and torch.cuda.is_available():
|
| 44 |
-
device_num = 0 # Cambia esto si tienes múltiples GPUs
|
| 45 |
-
else:
|
| 46 |
-
device_num = -1 # CPU
|
| 47 |
-
|
| 48 |
-
# Inicializar el pipeline de generación de texto
|
| 49 |
-
generator = pipeline('text-generation', model='gpt2', device=device_num)
|
| 50 |
-
|
| 51 |
-
# Generar el análisis
|
| 52 |
-
analysis = generator(prompt, max_length=max_length, num_return_sequences=1, temperature=0.7)[0]['generated_text']
|
| 53 |
-
|
| 54 |
-
return analysis
|
| 55 |
-
except Exception as e:
|
| 56 |
-
print(f"Error al generar el análisis: {e}")
|
| 57 |
-
return "Error al generar el análisis."
|
| 58 |
|
| 59 |
@gpu_decorator(duration=600) # Ajusta la duración según tus necesidades
|
| 60 |
def process_and_plot(
|
|
@@ -236,25 +214,16 @@ def process_and_plot(
|
|
| 236 |
buf.seek(0)
|
| 237 |
image = Image.open(buf)
|
| 238 |
|
| 239 |
-
# Crear el prompt para el análisis
|
| 240 |
prompt = f"""
|
| 241 |
-
|
| 242 |
-
|
| 243 |
-
|
| 244 |
-
|
| 245 |
-
|
| 246 |
-
|
| 247 |
-
|
| 248 |
-
|
| 249 |
-
|
| 250 |
-
|
| 251 |
-
**Producto:**
|
| 252 |
-
{product_results}
|
| 253 |
-
|
| 254 |
-
Analiza la calidad de cada modelo basándote en los valores de R² y RMSE. Proporciona un veredicto sobre la adecuación de los modelos y sugiere posibles mejoras o ajustes si es necesario.
|
| 255 |
-
"""
|
| 256 |
-
|
| 257 |
-
# Generar el análisis utilizando el modelo de lenguaje
|
| 258 |
analysis = generate_analysis(prompt, device=device)
|
| 259 |
|
| 260 |
return image, analysis
|
|
|
|
| 5 |
import matplotlib.pyplot as plt
|
| 6 |
from PIL import Image
|
| 7 |
import io
|
|
|
|
|
|
|
| 8 |
|
| 9 |
from bioprocess_model import BioprocessModel
|
| 10 |
from decorators import gpu_decorator # Asegúrate de que la ruta es correcta
|
|
|
|
| 28 |
|
| 29 |
@gpu_decorator(duration=300)
|
| 30 |
def generate_analysis(prompt, max_length=1024, device=None):
|
| 31 |
+
# Implementación existente para generar análisis usando Hugging Face o similar
|
| 32 |
+
# Por ejemplo, podrías usar un modelo de lenguaje para generar texto
|
| 33 |
+
# Aquí se deja como placeholder
|
| 34 |
+
analysis = "Análisis generado por el modelo de lenguaje."
|
| 35 |
+
return analysis
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
@gpu_decorator(duration=600) # Ajusta la duración según tus necesidades
|
| 38 |
def process_and_plot(
|
|
|
|
| 214 |
buf.seek(0)
|
| 215 |
image = Image.open(buf)
|
| 216 |
|
|
|
|
| 217 |
prompt = f"""
|
| 218 |
+
Eres un experto en modelado de bioprocesos.
|
| 219 |
+
Analiza los siguientes resultados experimentales y proporciona un veredicto sobre la calidad de los modelos, sugiriendo mejoras si es necesario.
|
| 220 |
+
Biomasa:
|
| 221 |
+
{biomass_results}
|
| 222 |
+
Sustrato:
|
| 223 |
+
{substrate_results}
|
| 224 |
+
Producto:
|
| 225 |
+
{product_results}
|
| 226 |
+
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 227 |
analysis = generate_analysis(prompt, device=device)
|
| 228 |
|
| 229 |
return image, analysis
|