Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 20 |
print(f"Using device: {device}")
|
| 21 |
|
| 22 |
# Load the Whisper model and processor
|
| 23 |
-
whisper_model_name = "openai/whisper-small
|
| 24 |
whisper_processor = WhisperProcessor.from_pretrained(whisper_model_name)
|
| 25 |
whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_name).to(device)
|
| 26 |
|
|
|
|
| 20 |
print(f"Using device: {device}")
|
| 21 |
|
| 22 |
# Load the Whisper model and processor
|
| 23 |
+
whisper_model_name = "openai/whisper-small"
|
| 24 |
whisper_processor = WhisperProcessor.from_pretrained(whisper_model_name)
|
| 25 |
whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_name).to(device)
|
| 26 |
|