Update app.py

app.py

@@ -24,9 +24,10 @@ def predict(model, image, detection_threshold: float = 0.5):
     return samples[0]["img"], preds[0]
     
 
-def show_preds(input_image):
+def show_preds(input_image, detection_threshold):
+    if detection_threshold==0: detection_threshold=0.5
 
-    img, pred = predict(model=model, image=input_image, detection_threshold=0.5)
+    img, pred = predict(model=model, image=input_image, detection_threshold)
     # print(pred)
     img = draw_pred(img=img, pred=pred, class_map=class_map, display_label=False, display_bbox=True)
     img = PIL.Image.fromarray(img)
@@ -45,7 +46,7 @@ description = "An object detection framework to localize regions under post-admi
 
 gr_interface = gr.Interface(
     fn=show_preds,
-    inputs=["image"],
+    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:")],
     title='Detect adaptive variants in admixed populations',
     description=description,