Update app.py

app.py

@@ -43,10 +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)
-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)
+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}")
 
 def getIP():
     ip_address = ''