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 |
|