Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -62,7 +62,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 62 |
|
| 63 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 64 |
import torch
|
| 65 |
-
from torch.utils.data import DataLoader
|
| 66 |
|
| 67 |
# Load model and tokenizer from Hugging Face Hub
|
| 68 |
MODEL_REPO = "ClimateDebunk/FineTunedDistilBert4SeqClass"
|
|
|
|
| 62 |
|
| 63 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
| 64 |
import torch
|
| 65 |
+
from torch.utils.data import DataLoader, Dataset
|
| 66 |
|
| 67 |
# Load model and tokenizer from Hugging Face Hub
|
| 68 |
MODEL_REPO = "ClimateDebunk/FineTunedDistilBert4SeqClass"
|