osanseviero
/

ConvTasNet_Libri1Mix_enhsingle_16k

Model card Files Files and versions Community

osanseviero commited on Aug 6, 2021

Commit

37a3159

•

1 Parent(s): 24403c3

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -11,7 +11,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("")
         self.sampling_rate = self.model.sample_rate
     def __call__(self, inputs: np.array) -> Tuple[np.array, int, List[str]]:

         # 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.sampling_rate = self.model.sample_rate
     def __call__(self, inputs: np.array) -> Tuple[np.array, int, List[str]]: