Update app.py
Browse files
app.py
CHANGED
|
@@ -151,7 +151,7 @@ def predict(cfg, sequences):
|
|
| 151 |
if cfg.task == "classification"
|
| 152 |
else model(inputs)
|
| 153 |
)
|
| 154 |
-
predictions
|
| 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 |
|