Update model.py

model.py

@@ -6,14 +6,15 @@ inference.
 from s3prl.downstream.runner import Runner
 from typing import Dict
 import torch
+import os
 
 
 class PreTrainedModel(Runner):
-    def __init__(self):
+    def __init__(self, path=""):
         """
         Loads model and tokenizer from local directory
         """
-        ckp_file = "hubert_asr.ckpt"
+        ckp_file = os.path.join(path, "hubert_asr.ckpt")
         ckp = torch.load(ckp_file, map_location='cpu')
         ckp["Args"].init_ckpt = ckp_file
         ckp["Args"].mode = "inference"