Update app.py

app.py

@@ -7,12 +7,8 @@ learn = from_pretrained_fastai("haripriyaram/Text-emotion-Recognizer-Model")
 # Prediction function
 def predict_emotion(text):
     pred_label, _, probs = learn.predict(text)
-
-    # # Handle nested vocab if needed
-    vocab = learn.dls.vocab[0] if isinstance(learn.dls.vocab[1], list) else learn.dls.vocab
-    probs_dict = {label: float(prob) for label, prob in zip(vocab, probs)}
-
-    return prediction, probs_dict
+    probs_dict = {label: float(prob) for label, prob in zip(learn.dls.vocab, probs)}
+    return pred_label, probs_dict
 
 # Gradio UI
 iface = gr.Interface(