Update app.py

app.py

@@ -23,6 +23,18 @@ def inference_sentence(text):
         predicted_class_id = logits.argmax().item()
     output = model.config.id2label[predicted_class_id]
     return output
+
+def frequencies(preds):
+	preds_dict = {"neutral": 0, "anger": 0, "fear": 0, "joy": 0, "love": 0, "sadness": 0}
+	for pred in preds:
+		preds_dict[pred] = preds_dict[pred] + 1
+	x = list(preds_dict.keys())
+	labels = list(preds_dict.values())
+
+	fig, ax = plt.subplots(figsize=(6, 6))
+	ax.pie(x, labels=labels, autopct='%.1f%%')
+	ax.set_title('Emotion frequencies')
+	return plt.tight_layout()
     
 def inference_dataset(file_object, option_list):
     tokenizer = AutoTokenizer.from_pretrained(inference_modelpath)
@@ -48,9 +60,9 @@ def inference_dataset(file_object, option_list):
         f.write(str(line[0]) + '\t' + str(line[1]) + '\t' + str(line[2]) + '\n')
     f.close()
     output1 = output
-    output2 = output3 = output4 = output5 = "This option was not selected."
+    output3 = output4 = output5 = "This option was not selected."
     if "emotion frequencies" in option_list:
-        output2 = "This option was selected."
+        output2 = frequencies(preds)
     if "emotion distribution over time" in option_list:
         output3 = "This option was selected."
     if "peaks" in option_list:
@@ -81,7 +93,7 @@ iface_dataset = gr.Interface(
             outputs = [
                 #gr.Textbox(label="Output file"),
                 "file",
-                gr.Textbox(label="Emotion frequencies"),
+                gr.Plot(label="Emotion frequencies"),
                 gr.Textbox(label="Emotion distribution over time"),
                 gr.Textbox(label="Peaks"),
                 gr.Textbox(label="Topics")