Update app.py

app.py

@@ -88,10 +88,10 @@ def classify_toxicity(audio_file, text_input, classify_anxiety, emo_class, expli
         # classification_output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
         classification_output = text_classifier(sequence_to_classify, candidate_labels, multi_label=True)
         print("class output ", type(classification_output))
-        classification_df = pd.DataFrame.from_dict(classification_output)
-        print("df ", classification_df.columns)
+        # classification_df = pd.DataFrame.from_dict(classification_output)
+        # print("df ", classification_df.columns)
 
-        plot.update(x=classification_df["labels"], y=classification_df["scores"])
+        # plot.update(x=classification_df["labels"], y=classification_df["scores"])
         
         return toxicity_score, classification_df, transcribed_text
         # return f"Toxicity Score ({available_models[selected_model]}): {toxicity_score:.4f}"
@@ -140,7 +140,7 @@ with gr.Blocks() as iface:
         submit_btn = gr.Button(label="Run")
     with gr.Column():
         out_val = gr.Textbox()
-        plot = gr.BarPlot()
+        plot = gr.Label()
         out_text = gr.Textbox()
     submit_btn.click(fn=classify_toxicity, inputs=[aud_input, text, anxiety_class, emo_class, explit_preference], outputs=[out_val, plot, out_text])