Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -34,11 +34,11 @@ phi_model = phi_model.merge_and_unload()
|
|
34 |
|
35 |
'''compute_type = 'float32'
|
36 |
if device != 'cpu':
|
37 |
-
compute_type = 'float16'''
|
38 |
|
39 |
audi_model = whisperx.load_model("small", device, compute_type='float16')
|
40 |
|
41 |
-
tokenizer = AutoTokenizer.from_pretrained('microsoft/phi-2'
|
42 |
tokenizer.pad_token = tokenizer.unk_token
|
43 |
|
44 |
|
|
|
34 |
|
35 |
'''compute_type = 'float32'
|
36 |
if device != 'cpu':
|
37 |
+
compute_type = 'float16'''
|
38 |
|
39 |
audi_model = whisperx.load_model("small", device, compute_type='float16')
|
40 |
|
41 |
+
tokenizer = AutoTokenizer.from_pretrained('microsoft/phi-2')
|
42 |
tokenizer.pad_token = tokenizer.unk_token
|
43 |
|
44 |
|