amaurisarchibald commited on
Commit
0e7b396
·
verified ·
1 Parent(s): bcf15a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"].scores[1])
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"