mizoru commited on
Commit
1654da2
·
1 Parent(s): d884a23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,9 +13,9 @@ learn = load_learner('xresnet50_pitch3.pkl')
13
 
14
  labels = learn.dls.vocab
15
 
16
- def predict(path_mic, path_upload):
17
- if path_upload: path = path_upload
18
- else: path = path_mic
19
  pred,pred_idx,probs = learn.predict(str(path))
20
 
21
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
13
 
14
  labels = learn.dls.vocab
15
 
16
+ def predict(Record, Upload):
17
+ if path_upload: path = Upload
18
+ else: path = Record
19
  pred,pred_idx,probs = learn.predict(str(path))
20
 
21
  return {labels[i]: float(probs[i]) for i in range(len(labels))}