Update model.py

model.py

@@ -2,6 +2,7 @@ import numpy as np
 
 from transformers import AutomaticSpeechRecognitionPipeline, AutoTokenizer, Wav2Vec2FeatureExtractor, Wav2Vec2ForCTC
 from typing import Dict
+from pathlib import Path
 
 from datasets import load_dataset
 
@@ -11,9 +12,10 @@ class PreTrainedModel():
         """
         Loads model and tokenizer from local directory
         """
-        model = Wav2Vec2ForCTC.from_pretrained(".")
-        tokenizer = AutoTokenizer.from_pretrained(".")
-        extractor = Wav2Vec2FeatureExtractor.from_pretrained(".")
+        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)
         
         self.model = AutomaticSpeechRecognitionPipeline(model=model, feature_extractor=extractor, tokenizer=tokenizer)