Update app.py

app.py

@@ -31,8 +31,9 @@ def show_preds(input_image, detection_threshold=0.5):
     img = draw_pred(img=img, pred=pred, class_map=class_map, display_label=False, display_bbox=True)
     img = PIL.Image.fromarray(img)
     pred_bbox = pred['bboxes']
+    pred_score = pred['scores']
     # print("Output Image: ", img.size, type(img))
-    return img, pred_bbox
+    return img, pred_bbox, pred_score
 
 # Populate examples in Gradio interface
 examples = [
@@ -46,7 +47,7 @@ description = "An object detection framework to localize regions under post-admi
 gr_interface = gr.Interface(
     fn=show_preds,
     inputs=[gr.Image(label="Upload 200x200 B&W image of ancestry-painted chromosomes:"), gr.Slider(minimum=0, maximum=1, step=0.1, default=0.5, label="Detection Threshold")],
-    outputs=[gr.Image(type="pil"), gr.Textbox(label="Predicted BBox:")],
+    outputs=[gr.Image(type="pil"), gr.Textbox(label="Predicted BBox:"), gr.Textbox(label="Predicted BBox Score:")],
     title='Detect adaptive variants in admixed populations',
     description=description,
     examples=examples)