Update app.py

app.py

@@ -9,7 +9,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from langchain.memory import ConversationBufferMemory
 
 # Configuración del modelo de lenguaje
-MODEL_NAME = "google/gemma-2b"  # Modelo más pequeño y eficiente
+MODEL_NAME = "PlanTL-GOB-ES/roberta-base-bne"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 HF_TOKEN = os.getenv("HF_TOKEN")  # Token de Hugging Face