Commit
·
cf435d8
1
Parent(s):
c2e5f66
add human
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ def predict_gender(filepath):
|
|
| 12 |
with torch.no_grad():
|
| 13 |
output = model.forward(audio)
|
| 14 |
probs = torch.softmax(output, dim=1)
|
| 15 |
-
prob_dict = {model.pred2gender[i]: float(prob) for i, prob in enumerate(probs[0])}
|
| 16 |
return prob_dict
|
| 17 |
|
| 18 |
audio_component = gr.Audio(type='filepath', label='Upload your audio file here')
|
|
|
|
| 12 |
with torch.no_grad():
|
| 13 |
output = model.forward(audio)
|
| 14 |
probs = torch.softmax(output, dim=1)
|
| 15 |
+
prob_dict = {'Human ' + model.pred2gender[i]: float(prob) for i, prob in enumerate(probs[0])}
|
| 16 |
return prob_dict
|
| 17 |
|
| 18 |
audio_component = gr.Audio(type='filepath', label='Upload your audio file here')
|