Update app.py

app.py

@@ -12,7 +12,8 @@ import pandas as pd
 
 from huggingface_hub import Repository
 from urllib.request import urlopen
-from transformers import AutoTokenizer, AutoModelWithLMHead 
+# from transformers import AutoTokenizer, AutoModelWithLMHead 
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 ## connection with HF datasets
 HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -43,7 +44,9 @@ SENTENCES_VALUE = """Raj loves Simran.\nLast year I lost my Dog.\nI bought a new
 cwd = os.getcwd()
 model_path = os.path.join(cwd)
 tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion")
-model_base = AutoModelWithLMHead.from_pretrained(model_path)
+# model_base = AutoModelWithLMHead.from_pretrained(model_path)
+# Instead of AutoModelWithLMHead
+model_base = AutoModelForCausalLM.from_pretrained(model_path)
 
 def getIP():
     ip_address = ''