Update models/speech_to_text/transcriber/model.py

models/speech_to_text/transcriber/model.py

@@ -1,4 +1,6 @@
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
+from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
+from transformers import pipeline
 
 def load_model_and_processor(model_name="openai/whisper-base"):
     """
@@ -11,6 +13,8 @@ def load_model_and_processor(model_name="openai/whisper-base"):
         model (WhisperForConditionalGeneration): Loaded Whisper model.
         processor (WhisperProcessor): Loaded processor for the model.
     """
-    model = WhisperForConditionalGeneration.from_pretrained(model_name)
-    processor = WhisperProcessor.from_pretrained(model_name)
+
+    processor = AutoProcessor.from_pretrained(model_name)
+    model = AutoModelForSpeechSeq2Seq.from_pretrained(model_name)
+    
     return model, processor