Update README.md

README.md

@@ -130,9 +130,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"):
+def download_model_from_hf(repo_id=model_name, model_dir="./"):
     """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"]
 
@@ -140,7 +139,7 @@ def download_model_from_hf(repo_id=model_name, model_dir="onnx"):
         hf_hub_download(
             repo_id=repo_id,
             filename=file,
-            local_dir=model_dir
+            local_dir=model_dir,
         )
     
     return files