devingulliver commited on
Commit
e669d58
·
verified ·
1 Parent(s): 19001c8

switch model to medium.en

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,8 +8,8 @@ import whisper
8
 
9
  os.makedirs("/home/user/.cache/whisper")
10
  urllib.request.urlretrieve(
11
- "https://openaipublic.azureedge.net/main/whisper/models/e5b1a55b89c1367dacf97e3e19bfd829a01529dbfdeefa8caeb59b3f1b81dadb/large-v3.pt",
12
- "/home/user/.cache/whisper/large-v3.pt"
13
  )
14
 
15
  writer = whisper.utils.get_writer("srt", "/dev/null")
@@ -17,7 +17,7 @@ writer = whisper.utils.get_writer("srt", "/dev/null")
17
  @spaces.GPU
18
  def generate(file, progress=gr.Progress(track_tqdm=True)):
19
  # get file to type bytes somehow
20
- model = whisper.load_model("large", device="cuda")
21
  audio = whisper.load_audio(file)
22
  result = model.transcribe(audio, verbose=False)
23
  out = io.StringIO()
 
8
 
9
  os.makedirs("/home/user/.cache/whisper")
10
  urllib.request.urlretrieve(
11
+ "https://openaipublic.azureedge.net/main/whisper/models/d7440d1dc186f76616474e0ff0b3b6b879abc9d1a4926b7adfa41db2d497ab4f/medium.en.pt",
12
+ "/home/user/.cache/whisper/medium.en.pt"
13
  )
14
 
15
  writer = whisper.utils.get_writer("srt", "/dev/null")
 
17
  @spaces.GPU
18
  def generate(file, progress=gr.Progress(track_tqdm=True)):
19
  # get file to type bytes somehow
20
+ model = whisper.load_model("medium.en", device="cuda")
21
  audio = whisper.load_audio(file)
22
  result = model.transcribe(audio, verbose=False)
23
  out = io.StringIO()