Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -52,6 +52,7 @@ def inference(input_image):
|
|
52 |
result = {}
|
53 |
for i in range(top5_prob.size(0)):
|
54 |
result[categories[top5_catid[i]]] = top5_prob[i].item()
|
|
|
55 |
return result
|
56 |
|
57 |
inputs = gr.inputs.Image(type='pil')
|
|
|
52 |
result = {}
|
53 |
for i in range(top5_prob.size(0)):
|
54 |
result[categories[top5_catid[i]]] = top5_prob[i].item()
|
55 |
+
print(result)
|
56 |
return result
|
57 |
|
58 |
inputs = gr.inputs.Image(type='pil')
|