Spaces:
Build error
Build error
Jason Adrian commited on
Commit ·
261f1b8
1
Parent(s): 70a8ae4
changes on app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ def image_classifier(inp):
|
|
| 31 |
result = model(inp)
|
| 32 |
|
| 33 |
# postprocess
|
| 34 |
-
labeled_result = {class_names[label]: confidence for label, confidence in zip(result.top5, result.top5conf)}
|
| 35 |
|
| 36 |
return labeled_result
|
| 37 |
|
|
|
|
| 31 |
result = model(inp)
|
| 32 |
|
| 33 |
# postprocess
|
| 34 |
+
labeled_result = {class_names[label]: confidence for label, confidence in zip(result.probs.top5, result.probs.top5conf)}
|
| 35 |
|
| 36 |
return labeled_result
|
| 37 |
|