64FC commited on
Commit
894def0
1 Parent(s): 20945f0

Change Whisper-small to Distil-Whisper-large

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -6,8 +6,14 @@ import gradio as gr
6
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
7
 
8
  # Load Whisper-small
 
 
 
 
 
 
9
  pipe = pipeline("automatic-speech-recognition",
10
- model="openai/whisper-small",
11
  device=device
12
  )
13
 
 
6
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
7
 
8
  # Load Whisper-small
9
+ #pipe = pipeline("automatic-speech-recognition",
10
+ # model="openai/whisper-small",
11
+ # device=device
12
+ #)
13
+
14
+ # Load Distil-Whisper-large
15
  pipe = pipeline("automatic-speech-recognition",
16
+ model="distil-whisper/distil-large-v2",
17
  device=device
18
  )
19