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 = v.draw_instance_predictions(outputs["instances"]
|
99 |
|
100 |
return out
|
101 |
|
|
|
95 |
outputs = predictor(im)
|
96 |
|
97 |
v = Visualizer(im, models[model_id]["metadata"], scale=1.2)
|
98 |
+
out = v.draw_instance_predictions(outputs["instances"].scores[1])
|
99 |
|
100 |
return out
|
101 |
|