Spaces:
Runtime error
Runtime error
Revert "Fix tensor accessing technique"
Browse filesThis reverts commit 9e88ff028e43f78286c3aead880fa154d0c3d30e.
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]
|
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:
|