Update app.py

app.py

@@ -49,7 +49,9 @@ def run_quality_check(dataset, column, n_samples):
     data = pl.read_parquet(f"hf://datasets/{dataset}@~parquet/{config}/train/0000.parquet", columns=[column])
     texts = data[column].to_list()
     predictions = predict(texts[:n_samples])
-    return pd.DataFrame({"quality": predictions})
+    counts = pd.DataFrame({"quality": predictions}).value_counts().to_frame()
+    counts.reset_index(inplace=True)
+    return counts
 
 
 with gr.Blocks() as demo: