Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- 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)
|