round the score to two instead of three dec places
Browse files
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"],
|
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,
|