Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -53,13 +53,6 @@ def inference(input_image):
|
|
53 |
for i in range(top5_prob.size(0)):
|
54 |
result[categories[top5_catid[i]]] = top5_prob[i].item()
|
55 |
print(result)
|
56 |
-
result = {
|
57 |
-
1:0.1,
|
58 |
-
2:0.2,
|
59 |
-
3:0.3,
|
60 |
-
4:0.4,
|
61 |
-
5:0
|
62 |
-
}
|
63 |
return result
|
64 |
|
65 |
inputs = gr.Image(type='pil')
|
|
|
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.Image(type='pil')
|