Hugging Face's logo

Spaces:
owaski-demo
/
Demo
Sleeping

App Files Community
chinmaydan commited on
Commit
2b2d3d3
·
1 Parent(s): 83dced6

Added Whisper model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
6
 
7
  processor = WhisperProcessor.from_pretrained("openai/whisper-large")
8
- model = SpeechT5ForSpeechToText.from_pretrained("openai/whisper-large")
9
 
10
  model.config.forced_decoder_ids = WhisperProcessor.get_decoder_prompt_ids(language="english", task="transcribe")
11
 
 
5
  from transformers import WhisperProcessor, WhisperForConditionalGeneration
6
 
7
  processor = WhisperProcessor.from_pretrained("openai/whisper-large")
8
+ model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large")
9
 
10
  model.config.forced_decoder_ids = WhisperProcessor.get_decoder_prompt_ids(language="english", task="transcribe")
11