sagawa commited on
Commit
4cbe855
1 Parent(s): a0912bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -151,7 +151,7 @@ def predict(cfg, sequences):
151
  if cfg.task == "classification"
152
  else model(inputs)
153
  )
154
- predictions.extend(preds.cpu().tolist())
155
 
156
  predictions = list(itertools.chain.from_iterable(predictions))
157
 
 
151
  if cfg.task == "classification"
152
  else model(inputs)
153
  )
154
+ predictions += preds.cpu().tolist()
155
 
156
  predictions = list(itertools.chain.from_iterable(predictions))
157