Hugging Face's logo

Spaces:
ClimateDebunk
/
submission-template
Sleeping

App Files Community
hbanduk commited on
Commit
9aaf0a5
·
verified ·
1 Parent(s): 4a96b36

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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"