hbanduk commited on
Commit
3240e8b
·
verified ·
1 Parent(s): bc7edfa

Update tasks/text.py

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