Update app.py

app.py

@@ -21,7 +21,7 @@ from shiny import App, ui, render, reactive, Session, module
 from detectron2.utils.visualizer import Visualizer, ColorMode
 from detectron2.data import Metadata
 
-from python_utils import load_model
+from python_utils import load_model, apply_nms
 # Load data and compute static values
 from shared import app_dir
 
@@ -41,6 +41,7 @@ app_ui = ui.page_fluid(
                          multiple=True,
                          accept=[".png", ".jpg", ".jpeg"]),
             ui.input_action_button("analyze", "Analyze", class_="btn-success"),
+            ui.input_switch("cls_agnostic_nms", "Class Agnostic NMS", False),
             width =300
         ),
         ui.output_ui("results_container"),
@@ -123,6 +124,10 @@ def server(input, output, session: Session):
 
                 # Run prediction with original BGR image
                 prediction = predictor(im)
+
+                if input.cls_agnostic_nms():
+                    prediction = apply_nms(prediction)
+                
                 classes = prediction["instances"].pred_classes.tolist()
 
                 results.append({