Spaces:

DroolingPanda
/

teachingAssistant

Running

Michael Hu commited on 11 days ago

Commit

6f92dbc

1 Parent(s): 6034fea

use unsloth whisper model to reduce the model size

Files changed (1) hide show

utils/stt.py CHANGED Viewed

@@ -62,7 +62,7 @@ class WhisperModel(ASRModel):
             use_safetensors=True
         ).to(self.device)
-        self.processor = AutoProcessor.from_pretrained("openai/whisper-large-v3")
         logger.info("Whisper model loaded successfully")
     def transcribe(self, audio_path):

             use_safetensors=True
         ).to(self.device)
+        self.processor = AutoProcessor.from_pretrained("unsloth/whisper-large-v3")
         logger.info("Whisper model loaded successfully")
     def transcribe(self, audio_path):