Norphel commited on
Commit
5900732
·
verified ·
1 Parent(s): 23f9512

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def generate_text(audio):
30
  input_dict = asr_processor(data, sampling_rate=sr, return_tensors="pt", padding=True)
31
 
32
  with torch.no_grad():
33
- outputs = model(**inputs).logits
34
  pred_ids = torch.argmax(outputs, dim=-1)[0]
35
 
36
  # Decode the prediction
 
30
  input_dict = asr_processor(data, sampling_rate=sr, return_tensors="pt", padding=True)
31
 
32
  with torch.no_grad():
33
+ outputs = asr_model(**inputs).logits
34
  pred_ids = torch.argmax(outputs, dim=-1)[0]
35
 
36
  # Decode the prediction