Update app.py

app.py

@@ -1,15 +1,16 @@
-import gradio as gr
-from fastai.text.all import load_learner
-from huggingface_hub import notebook_login
+from huggingface_hub import from_pretrained_fastai
 
-# Load FastAI model once
-learn = load_learner("emotion_classifier.pkl")
-learn.push_to_hub("fastai-emotion-classifier")
+# Load model directly from Hugging Face
+learn = from_pretrained_fastai("haripriyaram/Text-emotion-Recognizer-Model")
 
 # Prediction function
 def predict_emotion(text):
     pred_label, _, probs = learn.predict(text)
-    probs_dict = {label: float(prob) for label, prob in zip(learn.dls.vocab, probs)}
+
+    # Handle nested vocab if needed
+    vocab = learn.dls.vocab[0] if isinstance(learn.dls.vocab[0], list) else learn.dls.vocab
+    probs_dict = {label: float(prob) for label, prob in zip(vocab, probs)}
+
     return pred_label, probs_dict
 
 # Gradio UI