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