Update app.py

app.py

@@ -5,7 +5,7 @@ import time
 from functools import wraps
 import sys
 import spaces  # Asegúrate de que este módulo esté disponible y correctamente instalado
-from model import GenerRNA  # Importa tu modelo personalizado
+from model import GenerRNA  # Ahora que model.py existe, esto debería funcionar
 
 # Decorador para medir el tiempo de ejecución
 def medir_tiempo(func):
@@ -20,7 +20,6 @@ def medir_tiempo(func):
     return wrapper
 
 # Decorador para asignar GPU durante la ejecución de la función
-# Nota: Asegúrate de que el decorador @spaces.GPU esté disponible en tu entorno
 @spaces.GPU(duration=120)  # Asigna GPU durante 120 segundos
 @medir_tiempo
 def generar_rna_sequence(prompt, max_length=256):
@@ -71,7 +70,9 @@ except ValueError as e:
 # Cargar el modelo GenerRNA
 try:
     print("Cargando el modelo GenerRNA...")
-    model = GenerRNA()  # Instancia tu modelo personalizado
+    # Cargar la configuración del modelo si es necesario
+    config = GPT2Config.from_pretrained("tokenizer_bpe_1024")  # Ajusta el path si tienes una configuración específica
+    model = GenerRNA(config)  # Instancia tu modelo personalizado
     model.load_state_dict(torch.load("model.pt.recombined", map_location=device))
     model.to(device)
     model.eval()