Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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):
|