Update app.py

app.py

@@ -51,7 +51,11 @@ def vid_predict(media):
     pipeline = DetectionPipeline(
                 model=best_model,
                 image_processor=best_model._image_processor,
-                post_prediction_callback=best_model.get_post_prediction_callback(iou=0.25, conf=0.70),
+                post_prediction_callback=best_model.get_post_prediction_callback(iou=0.25, conf=0.70,
+                                                                                nms_top_k=100,  # Example value, adjust based on your needs
+                                                                                max_predictions=50,  # Example value, adjust based on your needs
+                                                                                multi_label_per_box=False,  # Example value, adjust based on your needs
+                                                                                class_agnostic_nms=False),
                 class_names=best_model._class_names,
             )