check meta

app.py

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

backend.py

@@ -90,6 +90,9 @@ def visualize_image(im, mode="BOTH", tree_threshold=0.7, building_threshold=0.7,
 
     dataset_names = MetadataCatalog.list()
     print(dataset_names)
+    metadata = MetadataCatalog.get(dataset_names[0])
+    print("metadata", type(metadata), metadata)
+    print('metadata.get("thing_classes")', type(metadata.get("thing_classes")), metadata.get("thing_classes"))
     # category_names = metadata.get("thing_classes")
     # visualizer = Visualizer(im[:, :, ::-1],
     #                         metadata=metadata,