Update app.py

app.py

@@ -156,7 +156,8 @@ def predict_emotion(context):
     emotion_prediction_pipeline = pipeline('text-classification', model=emotion_prediction_model, tokenizer=emotion_prediction_tokenizer, top_k=None)
     predictions = emotion_prediction_pipeline(context)
     emotion_scores = predictions[0]
-    emotion_pred = emotion_classes[np.argmax(emotion_scores)]
+    sorted_scores = sorted(emotion_scores.items(), key=lambda x: x[1], reverse=True)
+    emotion_pred = sorted_scores[0][0]
     return emotion_pred
 
 def generate_text(prompt, max_length=100, emotion=None):
@@ -202,4 +203,4 @@ with gr.Blocks() as demo:
     
     predict_btn.click(fn=lambda context: (predict_emotion(context), generate_response(context, emotion=predict_emotion(context))), inputs=context_input, outputs=[emotion_output, generated_text_output])
 
-demo.launch(share=True)
+demo.launch()