Update model.py

model.py

@@ -1,17 +1,15 @@
 import numpy as np
 from transformers import AutomaticSpeechRecognitionPipeline, AutoTokenizer, Wav2Vec2FeatureExtractor, Wav2Vec2ForCTC
 from typing import Dict
-from pathlib import Path
 
 class PreTrainedModel():
-    def __init__(self):
+    def __init__(self, path):
         """
         Loads model and tokenizer from local directory
         """
-        current_file_path = Path(__file__)
-        model = Wav2Vec2ForCTC.from_pretrained(current_file_path.parent)
-        tokenizer = AutoTokenizer.from_pretrained(current_file_path.parent)
-        extractor = Wav2Vec2FeatureExtractor.from_pretrained(current_file_path.parent)
+        model = Wav2Vec2ForCTC.from_pretrained(path)
+        tokenizer = AutoTokenizer.from_pretrained(path)
+        extractor = Wav2Vec2FeatureExtractor.from_pretrained(path)
         
         self.model = AutomaticSpeechRecognitionPipeline(model=model, feature_extractor=extractor, tokenizer=tokenizer)
     def __call__(self, inputs)-> Dict[str, str]: