setup app layout

app.py

@@ -1,7 +1,28 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def segment_image(input_image):
+    output_image = input_image
+    return output_image
+
+
+with gr.Blocks() as demo:
+
+    with gr.Row():
+        input_image = gr.Image(label='Input Image')
+        output_image = gr.Image(label='Output Image')
+
+    button = gr.Button('Mask Image')
+    button.click(segment_image, inputs=[input_image], outputs=output_image)
+
+    gr.Examples(
+        examples = [
+            ['./imgs/cat.jpg']
+        ],
+        inputs=[input_image],
+        outputs=[output_image],
+        fn=segment_image,
+        cache_examples=True
+    )
+
+demo.launch()

requirements.txt

@@ -0,0 +1,7 @@
+gradio
+opencv-python
+matplotlib
+numpy
+torch
+torchvision
+git+https://github.com/facebookresearch/segment-anything.git