Ahsen Khaliq commited on
Commit
aaae048
1 Parent(s): dd5f104

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def inference(img):
30
  result = {}
31
 
32
  for i in range(5):
33
- result[pred_names[i][1]] = str(int(100*pred_names[i][2])).item()
34
 
35
  return result
36
 
 
30
  result = {}
31
 
32
  for i in range(5):
33
+ result[pred_names[i][1]] = pred_names[i][2].item()
34
 
35
  return result
36