Ar4ikov commited on
Commit
833d68e
1 Parent(s): 3b87efa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,7 +27,7 @@ def predict(speech_array, sampling_rate):
27
  logits = model.to(device)(**inputs).logits
28
 
29
  scores = F.softmax(logits, dim=1).detach().cpu().numpy()[0]
30
- outputs = [{config.id2label[i]: f"{round(score * 100, 3):.1f}%"} for i, score in enumerate(scores)]
31
  return outputs
32
 
33
 
@@ -43,7 +43,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
43
 
44
  def transcribe(audio):
45
  sr, audio = audio[0], audio[1]
46
- return predict(audio, sr)[0]
47
 
48
 
49
  def get_asr_interface():
 
27
  logits = model.to(device)(**inputs).logits
28
 
29
  scores = F.softmax(logits, dim=1).detach().cpu().numpy()[0]
30
+ outputs = {config.id2label[i]: f"{round(score * 100, 3):.1f}%" for i, score in enumerate(scores)}
31
  return outputs
32
 
33
 
 
43
 
44
  def transcribe(audio):
45
  sr, audio = audio[0], audio[1]
46
+ return predict(audio, sr)
47
 
48
 
49
  def get_asr_interface():