fix

app.py

@@ -75,9 +75,9 @@ def plot_and_df(texts, preds):
     # counts.reset_index(inplace=True)
     return (
             gr.BarPlot(counts_df, x="quality", y="count", sort=None),
-            texts_df[texts_df["quality"] == "Low"][["text"]][:20],
-            texts_df[texts_df["quality"] == "Medium"][["text"]][:20],
-            texts_df[texts_df["quality"] == "High"][["text"]][:20],
+            texts_df[texts_df["quality"] == "Low"][["text"]][:min(texts_df.shape[0], 20)],
+            texts_df[texts_df["quality"] == "Medium"][["text"]][:min(texts_df.shape[0], 20)],
+            texts_df[texts_df["quality"] == "High"][["text"]][:min(texts_df.shape[0], 20)],
         )
 
 
@@ -99,6 +99,7 @@ def run_quality_check(dataset, config, split, column, batch_size, num_examples):
         return
 
     try:
+        logging.info(f"Loading hf://datasets/{dataset}@~parquet/{filename}")
         data = pl.read_parquet(f"hf://datasets/{dataset}@~parquet/{filename}", columns=[column])
     except Exception as error:
         yield f"❌ {error}", gr.BarPlot(), pd.DataFrame(), pd.DataFrame(), pd.DataFrame(), pd.DataFrame(),