fix

app.py

@@ -62,7 +62,7 @@ def run_quality_check(dataset, column, batch_size):
     texts = data[column].to_list()
     # batch_size = 100
     predictions, texts_processed = [], []
-    for i in range(5):
+    for i in range(0, batch_size*5, batch_size):
         batch_texts = texts[i:i+batch_size]
         batch_predictions = predict(batch_texts)
         predictions.extend(batch_predictions)