Update README.md

README.md

@@ -89,7 +89,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model_name = "skypro1111/mbart-large-50-verbalization"
 
-model = T5ForConditionalGeneration.from_pretrained(
+model = MBartForConditionalGeneration.from_pretrained(
         model_name,
         low_cpu_mem_usage=True,
         device_map=device,
@@ -130,7 +130,7 @@ 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, model_dir="onnx"):
     """Download ONNX models from HuggingFace Hub."""
     os.makedirs(model_dir, exist_ok=True)