Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from demo import automask_image_app, automask_video_app, sahi_autoseg_app
 
 
+
 def image_app():
     with gr.Blocks():
         with gr.Row():
@@ -57,6 +58,27 @@ def image_app():
             ],
             outputs=[output_image],
         )
+        
+        gr.Examples(
+            examples=[
+                [
+                    "testv1.jpg",
+                    "vit_l",
+                    16,
+                    64,
+                    0,
+                ],
+            
+            ],
+            inputs=[
+                seg_automask_image_file,
+                seg_automask_image_model_type,
+                seg_automask_image_points_per_side,
+                seg_automask_image_points_per_batch,
+                seg_automask_image_min_area,
+            ],  
+            outputs=[output_image],
+        )
 
 
 def video_app():
@@ -89,7 +111,7 @@ def video_app():
                                 value=16,
                                 label="Points per Side",
                             )
-                            
+
                             seg_automask_video_points_per_batch = gr.Slider(
                                 minimum=0,
                                 maximum=64,
@@ -114,6 +136,27 @@ def video_app():
             outputs=[output_video],
         )
 
+        # examples
+        gr.Examples(
+            examples=[
+                [
+                    "testv1.mp4",
+                    "vit_l",
+                    16,
+                    64,
+                    0,
+                ],  
+            ],
+            inputs=[
+                seg_automask_video_file,
+                seg_automask_video_model_type,
+                seg_automask_video_points_per_side,
+                seg_automask_video_points_per_batch,
+                seg_automask_video_min_area,
+            ],
+            outputs=[output_video],
+        )
+
 
 def sahi_app():
     with gr.Blocks():
@@ -121,33 +164,33 @@ def sahi_app():
             with gr.Column():
                 sahi_image_file = gr.Image(type="filepath").style(height=260)
                 sahi_autoseg_model_type = gr.Dropdown(
-                        choices=[
-                            "vit_h",
-                            "vit_l",
-                            "vit_b",
-                        ],
-                        value="vit_l",
-                        label="Sam Model Type",
-                    )
+                    choices=[
+                        "vit_h",
+                        "vit_l",
+                        "vit_b",
+                    ],
+                    value="vit_l",
+                    label="Sam Model Type",
+                )
 
                 with gr.Row():
                     with gr.Column():
                         sahi_model_type = gr.Dropdown(
-                                choices=[
-                                    "yolov5",
-                                    "yolov8",
-                                ],
-                                value="yolov5",
-                                label="Detector Model Type",
-                            )
+                            choices=[
+                                "yolov5",
+                                "yolov8",
+                            ],
+                            value="yolov5",
+                            label="Detector Model Type",
+                        )
                         sahi_image_size = gr.Slider(
                             minimum=0,
                             maximum=1600,
                             step=32,
                             value=640,
                             label="Image Size",
-                        ) 
-                            
+                        )
+
                         sahi_overlap_width = gr.Slider(
                             minimum=0,
                             maximum=1,
@@ -155,7 +198,7 @@ def sahi_app():
                             value=0.2,
                             label="Overlap Width",
                         )
-                        
+
                         sahi_slice_width = gr.Slider(
                             minimum=0,
                             maximum=640,
@@ -165,13 +208,13 @@ def sahi_app():
                         )
 
                     with gr.Row():
-                        with gr.Column():                          
+                        with gr.Column():
                             sahi_model_path = gr.Dropdown(
                                 choices=[
                                     "yolov5l.pt",
                                     "yolov5l6.pt",
                                     "yolov8l.pt",
-                                    "yolov8x.pt"
+                                    "yolov8x.pt",
                                 ],
                                 value="yolov5l6.pt",
                                 label="Detector Model Path",
@@ -183,7 +226,7 @@ def sahi_app():
                                 step=0.1,
                                 value=0.2,
                                 label="Confidence Threshold",
-                            )   
+                            )
                             sahi_overlap_height = gr.Slider(
                                 minimum=0,
                                 maximum=1,
@@ -216,29 +259,44 @@ def sahi_app():
                 sahi_slice_width,
                 sahi_overlap_height,
                 sahi_overlap_width,
-
             ],
             outputs=[output_image],
         )
+        
+        gr.Examples(
+            examples=[
+                [
+                    "testv2.jpg",
+                    "vit_l",
+                    "yolov5",
+                    "yolov5l6.pt",
+                    0.2,
+                    640,
+                    256,
+                    256,
+                    0.2,
+                    0.2,
+                ],
+            ],
+            inputs=[
+                sahi_image_file,
+                sahi_autoseg_model_type,
+                sahi_model_type,
+                sahi_model_path,
+                sahi_conf_th,
+                sahi_image_size,
+                sahi_slice_height,
+                sahi_slice_width,
+                sahi_overlap_height,
+                sahi_overlap_width,
+            ],
+            outputs=[output_image],
+        )
+
 
 def metaseg_app():
     app = gr.Blocks()
     with app:
-        gr.HTML(
-        """
-        <h1 style='text-align: center'>
-       MetaSeg: Segment Anything + Video + SAHI
-        </h1>
-        """
-    )
-        gr.HTML(
-            """
-            <h3 style='text-align: center'>
-            Follow me for more! 
-            <a href='https://twitter.com/kadirnar_ai' target='_blank'>Twitter</a> | <a href='https://github.com/kadirnar' target='_blank'>Github</a> | <a href='https://www.linkedin.com/in/kadir-nar/' target='_blank'>Linkedin</a>
-            </h3>
-            """
-    )
         with gr.Row():
             with gr.Column():
                 with gr.Tab("Image"):
@@ -247,7 +305,6 @@ def metaseg_app():
                     video_app()
                 with gr.Tab("SAHI"):
                     sahi_app()
-                
 
     app.queue(concurrency_count=1)
     app.launch(debug=True, enable_queue=True)