64FC commited on
Commit
44a183b
1 Parent(s): 6a07d2a

Change back to Whisper-small

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -6,14 +6,8 @@ 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
-
14
- # Load Distil-Whisper-large
15
  pipe = pipeline("automatic-speech-recognition",
16
- model="distil-whisper/distil-large-v2",
17
  device=device
18
  )
19
 
 
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