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