Initial Commit

app.py

@@ -17,7 +17,6 @@ def get_coords(evt: gr.SelectData):
     return f"{evt.index[0]}, {evt.index[1]}"
 
 def inference(image, input_label):
-    print(input_label)
     predictor.set_image(image)
     input_point = np.array([[int(input_label['label'].split(',')[0]), int(input_label['label'].split(',')[1])]])
     input_label = np.array([1])
@@ -39,11 +38,11 @@ with my_app:
         with gr.TabItem("Select your image"):
             with gr.Row():
                 with gr.Column():
-                    img_source = gr.Image(label="Please select picture.", value='./images/truck.jpg', shape=(768, 768))
+                    img_source = gr.Image(label="Please select picture.", value='./images/truck.jpg', shape=(1024, 1024))
                     coords = gr.Label(label="Image Coordinate")
                     infer = gr.Button(label="Segment")
                 with gr.Column():
-                    img_output = gr.Image(label="Output Mask")
+                    img_output = gr.Image(label="Output Mask", shape=(1024, 1024))
                     
         img_source.select(get_coords, [], coords)
         infer.click(