amaurisarchibald commited on
Commit
c4f26c3
·
verified ·
1 Parent(s): 06b9f2b

Update app.py

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