asquirous commited on
Commit
b5063ea
·
1 Parent(s): 9e88ff0

Revert "Fix tensor accessing technique"

Browse files

This reverts commit 9e88ff028e43f78286c3aead880fa154d0c3d30e.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def classify_image(img):
8
  img = PILImage.create(img)
9
  pred, idx, probs = learn.predict(img)
10
  output = dict(zip(labels, map(float, probs)))
11
- print(probs[0].item())
12
  for out in output:
13
  val = output[out]
14
  if val < 60:
 
8
  img = PILImage.create(img)
9
  pred, idx, probs = learn.predict(img)
10
  output = dict(zip(labels, map(float, probs)))
11
+ print(probs[0])
12
  for out in output:
13
  val = output[out]
14
  if val < 60: