Update app.py

app.py

@@ -4,6 +4,7 @@ import matplotlib
 import matplotlib.pyplot as plt
 import numpy as np
 import torch
+import glob
 import gradio as gr
 
 from PIL import Image
@@ -12,7 +13,7 @@ from segment_anything import SamAutomaticMaskGenerator, SamPredictor, sam_model_
 
 matplotlib.pyplot.switch_backend('Agg') # for matplotlib to work in gradio
 
-st.markdown(
+gr.markdown(
     """
 # Segment Anything Model (SAM) 
 ### A test on remote sensing data
@@ -71,5 +72,12 @@ with gr.Blocks() as demo:
 
     segment_image_button = gr.Button("Segment")
     segment_image_button.click(segment_image, inputs=[image], outputs=image_output)
-
+    gr.Examples(
+    [ 
+        glob.glob('./images/*')
+    ],
+    image,
+    image_output,
+    segment_image)
+    
 demo.launch()