Update app.py

app.py

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