mizoru commited on
Commit
a6cc630
·
1 Parent(s): 6d84a1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def predict(Record, Upload):
19
  if Upload: path = Upload
20
  else: path = Record
21
  spec,pred,pred_idx,probs = learn.predict(str(path), with_input=True)
22
- fig,ax = plt.subplots()
23
  show_image(spec, ax=ax)
24
  return [{labels[i]: float(probs[i]) for i in range(len(labels))}, fig]
25
 
 
19
  if Upload: path = Upload
20
  else: path = Record
21
  spec,pred,pred_idx,probs = learn.predict(str(path), with_input=True)
22
+ fig,ax = plt.subplots(figsize=(12,8))
23
  show_image(spec, ax=ax)
24
  return [{labels[i]: float(probs[i]) for i in range(len(labels))}, fig]
25