czkaiweb commited on
Commit
38fa861
·
1 Parent(s): eacbb93

Update app.py

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