davanstrien HF staff commited on
Commit
d46183f
1 Parent(s): ab438d6

catch all changes to toggle

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def format_row_for_model(row):
43
  int2label = {0: "new_dataset", 1: "not_new_dataset"}
44
 
45
 
46
- def get_predictions(data: list[dict], model=None, batch_size=64):
47
  if model is None:
48
  model = load_model()
49
  predictions = []
 
43
  int2label = {0: "new_dataset", 1: "not_new_dataset"}
44
 
45
 
46
+ def get_predictions(data: list[dict], model=None, batch_size=128):
47
  if model is None:
48
  model = load_model()
49
  predictions = []