sdafd commited on
Commit
c1d2f7d
·
verified ·
1 Parent(s): e8d8ed1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -12,13 +12,14 @@ logger = logging.getLogger("whisperx_app")
12
 
13
  # Device setup (force CPU)
14
  device = "cpu"
 
15
  torch.set_num_threads(os.cpu_count())
16
 
17
  # Pre-load models
18
  models = {
19
- "tiny": whisperx.load_model("tiny", device),
20
- "base": whisperx.load_model("base", device),
21
- "small": whisperx.load_model("small", device),
22
  }
23
 
24
  def transcribe(audio_file, model_size="base", debug=False):
 
12
 
13
  # Device setup (force CPU)
14
  device = "cpu"
15
+ compute_type = "int8"
16
  torch.set_num_threads(os.cpu_count())
17
 
18
  # Pre-load models
19
  models = {
20
+ "tiny": whisperx.load_model("tiny", device, compute_type=compute_type),
21
+ "base": whisperx.load_model("base", device, compute_type=compute_type),
22
+ "small": whisperx.load_model("small", device, compute_type=compute_type),
23
  }
24
 
25
  def transcribe(audio_file, model_size="base", debug=False):