Spaces:

dhruvsaxena11
/

Emotion_Recognition_in_Text

Sleeping

dhruvsaxena11 commited on Apr 9, 2024

Commit

5153048

verified ·

1 Parent(s): 2597bd8

Update emotion_recognition_fine_tuning.py

Files changed (1) hide show

emotion_recognition_fine_tuning.py CHANGED Viewed

@@ -17,7 +17,7 @@ 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("emotion_model_dhruv")
 loaded_tokenizer=AutoTokenizer.from_pretrained("google-bert/bert-base-uncased")
 def predict_emotion(text):

 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")
 loaded_tokenizer=AutoTokenizer.from_pretrained("google-bert/bert-base-uncased")
 def predict_emotion(text):