Spaces:
Runtime error
Runtime error
Update backend/app/train.py
Browse files- backend/app/train.py +1 -1
backend/app/train.py
CHANGED
|
@@ -5,7 +5,7 @@ from transformers import (
|
|
| 5 |
AutoModelForSequenceClassification,
|
| 6 |
TrainingArguments,
|
| 7 |
Trainer,
|
| 8 |
-
|
| 9 |
)
|
| 10 |
import torch
|
| 11 |
from datasets import Dataset
|
|
|
|
| 5 |
AutoModelForSequenceClassification,
|
| 6 |
TrainingArguments,
|
| 7 |
Trainer,
|
| 8 |
+
DataCollatorWithPadding,
|
| 9 |
)
|
| 10 |
import torch
|
| 11 |
from datasets import Dataset
|