piyushgrover commited on
Commit
701fc97
·
verified ·
1 Parent(s): 084272f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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', trust_remote_code=True)
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