Update tasks/text.py

tasks/text.py

@@ -65,7 +65,12 @@ async def evaluate_text(request: TextEvaluationRequest):
     import onnxruntime as ort
     
     # Load the ONNX model and tokenizer
-    MODEL_PATH = "./distilbert_quantized_dynamic.onnx"
+    from huggingface_hub import hf_hub_download
+
+    MODEL_REPO = "ClimateDebunk/Quantized_DistilBertForSequenceClassification"
+    MODEL_FILENAME = "distilbert_quantized_dynamic.onnx"
+    MODEL_PATH = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILENAME)
+
     tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
     ort_session = ort.InferenceSession(MODEL_PATH, providers=["CPUExecutionProvider"])