Update app.py

app.py

@@ -5,7 +5,7 @@ import os, sys, torch, json, csv, warnings, joblib, uuid
 os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
 import streamlit as st
 from pathlib import Path
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForSequenceClassification
 from unidecode import unidecode
 from datetime import datetime
 from huggingface_hub import hf_hub_download
@@ -288,8 +288,8 @@ if __name__ == '__main__':
 
     # Loading Saved Models  
     # Modelo Contexto
+    context_model = AutoModelForSequenceClassification.from_pretrained("tecuhtli/mori-context-model")    
     cont_tok = AutoTokenizer.from_pretrained("tecuhtli/mori-context-model")
-    context_model = AutoModelForSeq2SeqLM.from_pretrained("tecuhtli/mori-context-model")
     
     # Modelo Técnico
     tec_tok = AutoTokenizer.from_pretrained("tecuhtli/mori-tecnico-model")