aka7774 commited on
Commit
007739a
1 Parent(s): 99f2b87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from faster_whisper import WhisperModel
3
 
4
  model_size = 'large-v3'
5
- model = whisper.load_model(model_size, device="auto", compute_type="float16")
6
 
7
  # or run on GPU with INT8
8
  # model = WhisperModel(model_size, device="cuda", compute_type="int8_float16")
@@ -13,7 +13,7 @@ def speech_to_text(audio_file, _model_size):
13
  global model_size, model
14
  if model_size != _model_size:
15
  model_size = _model_size
16
- model = whisper.load_model(model_size, device="auto", compute_type="float16")
17
 
18
  with torch.no_grad():
19
  segments, info = model.transcribe(
 
2
  from faster_whisper import WhisperModel
3
 
4
  model_size = 'large-v3'
5
+ model = WhisperModel(model_size, device="auto", compute_type="float16")
6
 
7
  # or run on GPU with INT8
8
  # model = WhisperModel(model_size, device="cuda", compute_type="int8_float16")
 
13
  global model_size, model
14
  if model_size != _model_size:
15
  model_size = _model_size
16
+ model = WhisperModel(model_size, device="auto", compute_type="float16")
17
 
18
  with torch.no_grad():
19
  segments, info = model.transcribe(