Update app.py

app.py

@@ -139,30 +139,13 @@ 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=[
-        gr.Image(type="pil", label="Mask with Image"),
-        gr.Image(type="pil", label="Mask")
-        ],
-    title="SAM based Segment Annotator.",
-    description='Sketch the portion where you want to create Mask.',
-    examples=[
-        "./cardamage_example/0006.JPEG",
-        "./cardamage_example/0008.JPEG",
-        "./cardamage_example/0206.JPEG"
-    ]
-)
-demo.launch(enable_queue=False)
-'''
 
 with gr.Blocks() as demo:
-    gr.Markdown("# How to use")
+    gr.Markdown("# Segmentation-Annotator-using-MobileSAM")
     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.
-    """)
+    gr.Markdown("## Demo to run Mobile Segment Anything base model")
+    gr.Markdown("""This app uses the [MobileSAM](https://github.com/ChaoningZhang/MobileSAM.git) model to get a mask from a points in an image.""")
+    gr.Markdown("""Full code can be found here [SourceCode](https://github.com/ChaoningZhang/MobileSAM.git)""")
     with gr.Row():
         image_input = gr.Image(label="[Stroke] Draw on Image", tool='sketch',type='pil')
         image_output1 = gr.Image(type="pil", label="Mask with Image")