Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 learn = load_learner('nlp_model.pkl')
 
 labels = learn.dls.vocab
+emotion_labels = labels[1]
 
 examples = ["I can't believe you lied to me again! This is unacceptable!",
             "Got a surprise gift today, feeling overjoyed!"]
@@ -13,7 +14,7 @@ examples = ["I can't believe you lied to me again! This is unacceptable!",
 
 def classify_text(text):
     pred,pred_idx,probs = learn.predict(text)
-    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+    return {emotion_labels[i]: float(probs[i]) for i in range(len(emotion_labels))}
 
 interface = gr.Interface(
     fn=classify_text,