Jerich commited on
Commit
4279e53
·
verified ·
1 Parent(s): f12e9dc

Replaced the openai/whisper-tiny references with openai/whisper-small

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -61,13 +61,13 @@ class TalklasTranslator:
61
 
62
  def _initialize_stt_model(self):
63
  try:
64
- print("Trying to load openai/whisper-tiny...")
65
- self.stt_processor = WhisperProcessor.from_pretrained("openai/whisper-tiny")
66
- self.stt_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny")
67
  self.stt_model.to(self.device)
68
- print("Loaded openai/whisper-tiny successfully")
69
  except Exception as e:
70
- print(f"Failed to load openai/whisper-tiny: {e}")
71
  print("Falling back to openai/whisper-base...")
72
  try:
73
  self.stt_processor = WhisperProcessor.from_pretrained("openai/whisper-base")
 
61
 
62
  def _initialize_stt_model(self):
63
  try:
64
+ print("Trying to load openai/whisper-small...")
65
+ self.stt_processor = WhisperProcessor.from_pretrained("openai/whisper-small")
66
+ self.stt_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small")
67
  self.stt_model.to(self.device)
68
+ print("Loaded openai/whisper-small successfully")
69
  except Exception as e:
70
+ print(f"Failed to load openai/whisper-small: {e}")
71
  print("Falling back to openai/whisper-base...")
72
  try:
73
  self.stt_processor = WhisperProcessor.from_pretrained("openai/whisper-base")