Spaces:
Paused
Paused
Replaced the openai/whisper-tiny references with openai/whisper-small
Browse files
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-
|
| 65 |
-
self.stt_processor = WhisperProcessor.from_pretrained("openai/whisper-
|
| 66 |
-
self.stt_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-
|
| 67 |
self.stt_model.to(self.device)
|
| 68 |
-
print("Loaded openai/whisper-
|
| 69 |
except Exception as e:
|
| 70 |
-
print(f"Failed to load openai/whisper-
|
| 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")
|