polinaeterna HF staff commited on
Commit
8aeb50a
·
1 Parent(s): 29a8f74
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -115,9 +115,9 @@ def run_quality_check(dataset, config, split, column, batch_size, num_examples):
115
  batch_predictions = predict(batch_texts)
116
  predictions.extend(batch_predictions)
117
  texts_processed.extend(batch_texts)
118
- yield {"check in progress...": i / num_examples}, *plot_and_df(texts_processed, predictions), pd.DataFrame()
119
 
120
- yield {"finished": 1.}, *plot_and_df(texts_processed, predictions), data_sample
121
 
122
 
123
  PERSPECTIVE_API_KEY = os.environ.get("PERSPECTIVE_API_KEY")
 
115
  batch_predictions = predict(batch_texts)
116
  predictions.extend(batch_predictions)
117
  texts_processed.extend(batch_texts)
118
+ yield {"quality check in progress...": i / num_examples}, *plot_and_df(texts_processed, predictions), pd.DataFrame()
119
 
120
+ yield {"quality check finished": 1.}, *plot_and_df(texts_processed, predictions), data_sample
121
 
122
 
123
  PERSPECTIVE_API_KEY = os.environ.get("PERSPECTIVE_API_KEY")