add plot

app.py

@@ -51,7 +51,7 @@ def run_quality_check(dataset, column, n_samples):
     predictions = predict(texts[:n_samples])
     counts = pd.DataFrame({"quality": predictions}).value_counts().to_frame()
     counts.reset_index(inplace=True)
-    return counts
+    return gr.BarPlot(counts, x="quality", y="count")
 
 
 with gr.Blocks() as demo:
@@ -83,9 +83,8 @@ with gr.Blocks() as demo:
     text_column = gr.Textbox(placeholder="text", label="Text colum name to check (data must be non-nested, raw texts!)")
     n_samples = gr.Number(label="Num first samples to run check")
     gr_check_btn = gr.Button("Check Dataset")
-    # plot = gr.BarPlot()
-    df = gr.DataFrame()
-    gr_check_btn.click(run_quality_check, inputs=[dataset_name, text_column, n_samples], outputs=[df])
-    # gr.BarPlot(df)
+    plot = gr.BarPlot()
+    # df = gr.DataFrame(visible=False)
+    gr_check_btn.click(run_quality_check, inputs=[dataset_name, text_column, n_samples], outputs=[plot])
 
 demo.launch()