Update app.py
Browse files
app.py
CHANGED
@@ -95,7 +95,7 @@ def inferenceScore(image_url, image, min_score, model_name):
|
|
95 |
outputs = predictor(im)
|
96 |
|
97 |
v = Visualizer(im, models[model_id]["metadata"], scale=1.2)
|
98 |
-
out =
|
99 |
return float(out.item())
|
100 |
|
101 |
title = "# DBMDZ Detectron2 Model Demo"
|
|
|
95 |
outputs = predictor(im)
|
96 |
|
97 |
v = Visualizer(im, models[model_id]["metadata"], scale=1.2)
|
98 |
+
out = outputs["instances"].scores[1]
|
99 |
return float(out.item())
|
100 |
|
101 |
title = "# DBMDZ Detectron2 Model Demo"
|