Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ categories = learn.dls.vocab
|
|
| 13 |
|
| 14 |
def tensor2labels(img):
|
| 15 |
output = learn.predict(img)
|
| 16 |
-
index = torch.where(output[1].sigmoid()>0.
|
| 17 |
category = categories[index]
|
| 18 |
print(category)
|
| 19 |
if 'dog' in category:
|
|
|
|
| 13 |
|
| 14 |
def tensor2labels(img):
|
| 15 |
output = learn.predict(img)
|
| 16 |
+
index = torch.where(output[1].sigmoid()>0.4)[0]
|
| 17 |
category = categories[index]
|
| 18 |
print(category)
|
| 19 |
if 'dog' in category:
|