ongkn commited on
Commit
b792534
1 Parent(s): 8cc08e3

round the score to two instead of three dec places

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def classify_image(input):
44
  )
45
  result = gradCam.get_top_category(model, tensorResized)
46
  cls = result[0]["label"]
47
- result[0]["score"] = round(result[0]["score"], 3)
48
  clsIdx = gradCam.category_name_to_index(model, cls)
49
  clsTarget = ClassifierOutputTarget(clsIdx)
50
  gradCamImage = gradCam.run_grad_cam_on_image(model=model,
 
44
  )
45
  result = gradCam.get_top_category(model, tensorResized)
46
  cls = result[0]["label"]
47
+ result[0]["score"] = round(result[0]["score"], 2)
48
  clsIdx = gradCam.category_name_to_index(model, cls)
49
  clsTarget = ClassifierOutputTarget(clsIdx)
50
  gradCamImage = gradCam.run_grad_cam_on_image(model=model,