Spaces:

poltextlab
/

mores_pulse

Running

kovacsvi commited on 12 days ago

Commit

ba0fb9c

1 Parent(s): 226b684

oops

Files changed (1) hide show

app.py CHANGED Viewed

@@ -230,7 +230,7 @@ def predict_wrapper(text, language):
     print(results)
     print(results_heatmap)
-    figure = plot_average_emotion_barplot(prepare_heatmap_data(results_heatmap))
     heatmap = plot_emotion_heatmap(prepare_heatmap_data(results_heatmap))
     output_info = f'Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model. '
     funding_info = "The research was funded by European Union’s Horizon 2020 research and innovation program, “MORES” project (Grant No.: 101132601)"

     print(results)
     print(results_heatmap)
+    figure = plot_average_emotion_barplot(results_heatmap)
     heatmap = plot_emotion_heatmap(prepare_heatmap_data(results_heatmap))
     output_info = f'Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model. '
     funding_info = "The research was funded by European Union’s Horizon 2020 research and innovation program, “MORES” project (Grant No.: 101132601)"