hbanduk commited on
Commit
a0f5eb6
·
verified ·
1 Parent(s): b653c17

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +1 -2
tasks/text.py CHANGED
@@ -63,10 +63,9 @@ async def evaluate_text(request: TextEvaluationRequest):
63
  from transformers import DistilBertTokenizer
64
  import numpy as np
65
  import onnxruntime as ort
 
66
 
67
  # Load the ONNX model and tokenizer
68
- from huggingface_hub import hf_hub_download
69
-
70
  MODEL_REPO = "ClimateDebunk/Quantized_DistilBertForSequenceClassification"
71
  MODEL_FILENAME = "distilbert_quantized_dynamic.onnx"
72
  MODEL_PATH = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILENAME)
 
63
  from transformers import DistilBertTokenizer
64
  import numpy as np
65
  import onnxruntime as ort
66
+ from huggingface_hub import hf_hub_download
67
 
68
  # Load the ONNX model and tokenizer
 
 
69
  MODEL_REPO = "ClimateDebunk/Quantized_DistilBertForSequenceClassification"
70
  MODEL_FILENAME = "distilbert_quantized_dynamic.onnx"
71
  MODEL_PATH = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILENAME)