Update onnx/infer_onnx_hf.py

onnx/infer_onnx_hf.py

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