Update app.py

app.py

@@ -35,6 +35,10 @@ def point_selection(mask_sim, topk=1):
     
     return topk_xy, topk_label, last_xy, last_label
 
+def reset_data():
+    global cache_data
+    cache_data = None
+
 def inference_scribble(image):
     # in context image and mask
     ic_image = image["image"]
@@ -135,6 +139,7 @@ def inference_scribble(image):
     return [Image.fromarray((mask_colors * 0.6 + test_image * 0.4).astype('uint8'), 'RGB'), 
             Image.fromarray((mask_colors ).astype('uint8'), 'RGB')]
 
+'''
 demo = gr.Interface(fn=inference_scribble,
     inputs=gr.Image(label="[Stroke] Draw on Image", tool='sketch',type='pil'),
     outputs=[
@@ -149,4 +154,26 @@ demo = gr.Interface(fn=inference_scribble,
         "./cardamage_example/0206.JPEG"
     ]
 )
-demo.launch(enable_queue=False)
+demo.launch(enable_queue=False)
+'''
+
+with gr.Blocks() as demo:
+    gr.Markdown("# How to use")
+    gr.Markdown("To start, input an image, then use the brush to create dots on the object which you want to segment, don't worry if your dots aren't perfect as the code will find the middle of each drawn item. Then press the segment button to create masks for the object that the dots are on.")
+    gr.Markdown("# Demo to run Segment Anything base model")
+    gr.Markdown("""This app uses the [Segment Anything](https://huggingface.co/facebook/sam-vit-base) model from Meta to get a mask from a points in an image.
+    """)
+    with gr.Tab("Flip Image"):
+        with gr.Row():
+            image_input = gr.Image(label="[Stroke] Draw on Image", tool='sketch',type='pil')
+            image_output = [
+        gr.Image(type="pil", label="Mask with Image"),
+        gr.Image(type="pil", label="Mask")
+        ],
+        
+        image_button = gr.Button("Genarate-Segment-Mask")
+
+    image_button.click(inference_scribble, inputs=image_input, outputs=image_output)
+    image_input.upload(reset_data)
+
+demo.launch()