Update app.py

app.py

@@ -77,6 +77,27 @@ def inference(image_url, image, min_score, model_name):
 
     return out.get_image()
 
+def inferenceScore(image_url, image, min_score, model_name):
+    if image_url:
+        r = requests.get(image_url)
+        if r:
+            im = np.frombuffer(r.content, dtype="uint8")
+            im = cv2.imdecode(im, cv2.IMREAD_COLOR)
+    else:
+        # Model expect BGR!
+        im = image[:,:,::-1]
+
+    model_id = model_name_to_id[model_name]
+
+    models[model_id]["cfg"].MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
+    predictor = DefaultPredictor(models[model_id]["cfg"])
+
+    outputs = predictor(im)
+
+    v = Visualizer(im, models[model_id]["metadata"], scale=1.2)
+    out = v.draw_instance_predictions(outputs["instances"]..scores[1])
+
+    return out
 
 title = "# DBMDZ Detectron2 Model Demo"
 description = """
@@ -107,7 +128,7 @@ with gr.Blocks() as demo:
 
     inference_button = gr.Button("Submit")
     
-    inference_button.click(fn=inference, inputs=[url_input, image_input, min_score, model_name], outputs=output_image)
+    inference_button.click(fn=inferenceScore, inputs=[url_input, image_input, min_score, model_name], outputs=output_image)
 
     gr.Markdown(footer)