[feature] add point-per-side config

app.py

@@ -22,8 +22,12 @@ hourglass_args = {
     },
 }
 
-def predict(image, speed_mode):
-    mask_generator = SamAutomaticMaskGenerator(build_sam(checkpoint="sam_vit_h_4b8939.pth", hourglass_kwargs=hourglass_args[speed_mode]))
+def predict(image, speed_mode, point_per_side):
+    mask_generator = SamAutomaticMaskGenerator(
+        build_sam(checkpoint="sam_vit_h_4b8939.pth", hourglass_kwargs=hourglass_args[speed_mode]), 
+        point_per_side=point_per_side,
+        points_per_batch=64 if point_per_side > 12 else point_per_side * point_per_side
+    )
     masks = mask_generator.generate(image)
 
     if len(masks) == 0:
@@ -54,6 +58,11 @@ def main():
             with gr.Row():
                 with gr.Column():
                     input_image = gr.Image(label="Input Image")
+                    points_per_side = gr.Dropdown(
+                        choices=[4, 6, 8, 12, 16, 32],
+                        value=12, 
+                        label="Points per Side",
+                    )
                     speed_mode = gr.Dropdown(
                         choices=list(hourglass_args.keys()),
                         value="baseline", 
@@ -77,7 +86,7 @@ def main():
         
         run_btn.click(
             fn=predict, 
-            inputs=[input_image, speed_mode], 
+            inputs=[input_image, speed_mode, points_per_side], 
             outputs=output_image
         )
         clear_btn.click(