Update app.py

app.py

@@ -13,7 +13,8 @@ from langchain_core.prompts import ChatPromptTemplate, FewShotChatMessagePromptT
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnableLambda
 from datetime import date
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
+
 import threading
 import time
 # Environment variables
@@ -28,7 +29,7 @@ def load_model():
     model_name = "distilbert-base-uncased"
 
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
+    model = AutoModelForSequenceClassification.from_pretrained(model_name)
     return model, tokenizer
 def load_from_pickle(filename):
     with open(filename, "rb") as file: