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