Update app.py

app.py

@@ -6,11 +6,14 @@ from fastai.text.all import load_learner
 # Load the trained model
 learn = load_learner('emotion_classifier.pkl')
 
-# Define the prediction function
+
 def predict_emotion(text):
-    pred, pred_idx, probs = learn.predict(text)
-    confidence = float(probs[learn.dls.vocab[1].o2i[pred]])
-    return f"{pred} (confidence: {confidence:.2f})"
+    try:
+        pred, pred_idx, probs = learn.predict(text)
+        confidence = float(probs[pred_idx])
+        return {"label": pred, "confidence": confidence}
+    except Exception as e:
+        return {"error": str(e)}
 
 iface = gr.Interface(
     fn=predict_emotion,