Spaces:
Build error
Build error
Update app.py
Browse files
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]:
|
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 |
|