Rename app_old.py to app.py

app_old.py → app.py

@@ -13,6 +13,33 @@ sam = sam_model_registry[model_type](checkpoint="./tinysam/tinysam.pth")
 
 predictor = SamPredictor(sam)
 
+examples = [
+    ["assets/1.jpg"],
+    ["assets/2.jpg"],
+    ["assets/3.jpg"],
+    ["assets/4.jpeg"],
+    ["assets/5.jpg"],
+    ["assets/6.jpeg"]
+]
+
+default_example = examples[0]
+# Description
+title = "<center><strong><font size='8'>TinySAM<font></strong> <a href='https://github.com/xinghaochen/TinySAM'><font size='6'>[GitHub]</font></a> </center>"
+description_p = """# Interactive Instance Segmentation
+                - Point-prompt instruction
+                <ol>
+                <li> Click on the left image (point input), visualizing the point on the right image </li>
+                <li> Click the button of Segment with Point Prompt </li>
+                </ol>
+                - Box-prompt instruction
+                <ol>
+                <li> Click on the left image (one point input), visualizing the point on the right image </li>
+                <li> Click on the left image (another point input), visualizing the point and the box on the right image</li>
+                <li> Click the button of Segment with Box Prompt </li>
+                </ol>
+                - Github [link](https://github.com/xinghaochen/TinySAM)
+              """
+
 def infer(img):
   if img is None:
       gr.Error("Please upload an image and select a point.")
@@ -54,6 +81,18 @@ with gr.Blocks() as demo:
                 type="pil"
             )
         output = gr.AnnotatedImage()
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("Try some of the examples below ⬇️")
+            gr.Examples(
+                examples=examples,
+                inputs=[im],
+                examples_per_page=4,
+            )
+
+        with gr.Column():
+            # Description
+            gr.Markdown(description_p)
     im.change(infer, inputs=im, outputs=output)
 
 demo.launch(debug=True)