Ar4ikov commited on
Commit
89a1937
1 Parent(s): 55c2d79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def predict(speech_array, sampling_rate):
33
  logits = model.to(device)(**inputs).logits
34
 
35
  scores = F.softmax(logits, dim=1).detach().cpu().numpy()[0]
36
- outputs = {config.id2label[i]: f"{round(score * 100, 3):.1f}%" for i, score in enumerate(scores)}
37
  return outputs
38
 
39
 
 
33
  logits = model.to(device)(**inputs).logits
34
 
35
  scores = F.softmax(logits, dim=1).detach().cpu().numpy()[0]
36
+ outputs = {config.id2label[i]: round(score * 100, 3) for i, score in enumerate(scores)}
37
  return outputs
38
 
39