asquirous commited on
Commit
f7f9f63
·
1 Parent(s): 641fe93

Fix type error

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,10 +9,10 @@ def classify_image(img):
9
  img = PILImage.create(img)
10
  pred, idx, probs = learn.predict(img)
11
  output = dict(zip(labels, map(float, probs)))
12
- print(probs[0]*100)
13
  for out in output:
14
  val = output[out]
15
- if val.item()*100 < 60:
16
  return {"Not Sure/Others": 0}
17
  return output
18
 
 
9
  img = PILImage.create(img)
10
  pred, idx, probs = learn.predict(img)
11
  output = dict(zip(labels, map(float, probs)))
12
+ print(probs[0].item()*100)
13
  for out in output:
14
  val = output[out]
15
+ if val*100 < 60:
16
  return {"Not Sure/Others": 0}
17
  return output
18