Update pipeline.py

pipeline.py

@@ -3,6 +3,7 @@ from typing import Dict, List, Tuple
 import numpy as np
 from asteroid import separate
 from asteroid.models import BaseModel
+import os
 
 
 class PreTrainedPipeline():
@@ -11,7 +12,7 @@ class PreTrainedPipeline():
         # Preload all the elements you are going to need at inference.
         # For instance your model, processors, tokenizer that might be needed.
         # This function is only called once, so do all the heavy processing I/O here"""
-        self.model = BaseModel.from_pretrained(path)
+        self.model = BaseModel.from_pretrained(os.path.join(path, "pytorch_model.bin"))
         self.sampling_rate = self.model.sample_rate
 
     def __call__(self, inputs: np.array) -> Tuple[np.array, int, List[str]]: