Update app.py

app.py

@@ -6,10 +6,10 @@ import numpy as np
 import torch
 import glob
 import gradio as gr
-
 from PIL import Image
 
 from segment_anything import SamAutomaticMaskGenerator, SamPredictor, sam_model_registry
+from torchvision.utils import draw_segmentation_masks
 
 matplotlib.pyplot.switch_backend('Agg') # for matplotlib to work in gradio
 
@@ -74,7 +74,8 @@ with gr.Blocks() as demo:
     segment_image_button.click(segment_image, inputs=[image], outputs=image_output)
     gr.Examples(
     [ 
-        glob.glob('./images/*')
+        # glob.glob('./images/*')
+        './images/test (1).jpg'
     ],
     image,
     image_output,