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"
|