textToSQL commited on
Commit
3f3a4e9
·
1 Parent(s): f842c3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -1,7 +1,13 @@
1
  import whisper
2
  import gradio as gr
3
 
4
- model = whisper.load_model("large-v2")
 
 
 
 
 
 
5
 
6
  def transcribe(audio):
7
 
 
1
  import whisper
2
  import gradio as gr
3
 
4
+ # load model and processor
5
+
6
+ processor = WhisperProcessor.from_pretrained("openai/whisper-medium")
7
+
8
+ model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-medium")
9
+
10
+ model.config.forced_decoder_ids = None
11
 
12
  def transcribe(audio):
13