Update onnx/infer_onnx_hf.py

onnx/infer_onnx_hf.py

@@ -8,7 +8,7 @@ from huggingface_hub import hf_hub_download
 model_name = "skypro1111/mbart-large-50-verbalization"
 
 
-def download_model_from_hf(repo_id=model_name):
+def download_model_from_hf(repo_id=model_name, model_dir="./"):
     """Download ONNX models from HuggingFace Hub."""
     
     files = ["onnx/encoder_model.onnx", "onnx/decoder_model.onnx", "onnx/decoder_model.onnx_data"]
@@ -17,6 +17,7 @@ def download_model_from_hf(repo_id=model_name):
         hf_hub_download(
             repo_id=repo_id,
             filename=file,
+            local_dir=model_dir,
         )
     
     return files