Shubham09 commited on
Commit
fac7d33
·
1 Parent(s): 6d5c6c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -33,9 +33,10 @@ def asr_transcript(input_file):
33
 
34
  speech = load_data(input_file)
35
  #Tokenize
36
- input_values = tokenizer(speech, return_tensors="pt").input_values
 
37
  #Take logits
38
- logits = model(input_values).logits
39
  #Take argmax
40
  predicted_ids = torch.argmax(logits, dim=-1)
41
  #Get the words from predicted word ids
 
33
 
34
  speech = load_data(input_file)
35
  #Tokenize
36
+ input_features = processor(speech, return_tensors="pt").input_features #, padding="longest" , return_tensors="pt"
37
+ #input_values = tokenizer(speech, return_tensors="pt").input_values
38
  #Take logits
39
+ logits = model(input_features).logits
40
  #Take argmax
41
  predicted_ids = torch.argmax(logits, dim=-1)
42
  #Get the words from predicted word ids