czkaiweb commited on
Commit
6796683
·
1 Parent(s): 6a31e95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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')