mizoru commited on
Commit
114ae09
·
1 Parent(s): 0191af1

Update app.py

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