Update app.py

app.py

@@ -10,14 +10,18 @@ Original file is located at
 
 """Our Application"""
 
-from transformers import TFBertForSequenceClassification,AutoTokenizer
 
 import numpy as np
 
 import tensorflow as tf # Apply softmax using tf.nn.softmax
 
 # Load the fine-tuned model from the saved directory
-loaded_model = TFBertForSequenceClassification.from_pretrained("https://huggingface.co/spaces/dhruvsaxena11/Emotion_Recognition_in_Text/blob/main/tf_model.h5")
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("dhruvsaxena11/emoton_model_dhruv")
+loaded_model = AutoModelForSequenceClassification.from_pretrained("dhruvsaxena11/emoton_model_dhruv")
+# loaded_model = TFBertForSequenceClassification.from_pretrained("https://huggingface.co/spaces/dhruvsaxena11/Emotion_Recognition_in_Text/blob/main/tf_model.h5")
 loaded_tokenizer=AutoTokenizer.from_pretrained("google-bert/bert-base-uncased")
 
 def predict_emotion(text):