change gradio

app.py

@@ -3,11 +3,11 @@ from backend import visualize_image
 
 # gradio inputs
 image_input = gr.components.Image(type="pil", label="Input Image")
-color_mode_select = gr.components.Radio(["Black/white", "Random", "Segmentation"], label="Color Mode", default="Segmentation")
-mode_dropdown = gr.components.Dropdown(["Trees", "Buildings", "Both"], label="Detection Mode", default="Both")
+color_mode_select = gr.components.Radio(["Black/white", "Random", "Segmentation"], label="Color Mode")
+mode_dropdown = gr.components.Dropdown(["Trees", "Buildings", "Both"], label="Detection Mode")
 
-tree_threshold_slider = gr.components.Slider(0, 1, 0.1, 0.7, label='Set confidence threshold "%" for trees')
-building_threshold_slider = gr.components.Slider(0, 1, 0.1, 0.7, label='Set confidence threshold "%" for buildings')
+tree_threshold_slider = gr.components.Slider(minimum=0, maximum=1, step=0.1, default=0.7, label='Set confidence threshold "%" for trees')
+building_threshold_slider = gr.components.Slider(minimum=0, maximum=1, step=0.1, default=0.7, label='Set confidence threshold "%" for buildings')
 
 # gradio outputs
 output_image = gr.components.Image(type="pil", label="Output Image")

backend.py

@@ -65,10 +65,10 @@ def map_color_mode(color_mode):
         return ColorMode.IMAGE_BW
     elif color_mode == "Random":
         return ColorMode.IMAGE
-    elif color_mode == "Segmentation":
+    elif color_mode == "Segmentation" or color_mode == None:
         return ColorMode.SEGMENTATION
 
-def visualize_image(im, mode="BOTH", tree_threshold=0.7, building_threshold=0.7, color_mode=ColorMode.SEGMENTATION):
+def visualize_image(im, mode, tree_threshold:float, building_threshold:float, color_mode):
     im = np.array(im)
     color_mode = map_color_mode(color_mode)
 
@@ -76,7 +76,7 @@ def visualize_image(im, mode="BOTH", tree_threshold=0.7, building_threshold=0.7,
         instances = segment_tree(im, tree_threshold)
     elif mode == "Buildings":
         instances = segment_building(im, building_threshold)
-    elif mode == "Both":
+    elif mode == "Both" or mode == None:
         tree_instances = segment_tree(im, tree_threshold)
         building_instances = segment_building(im, building_threshold)
         instances = Instances.cat([tree_instances, building_instances])