input

app.py

@@ -8,7 +8,7 @@ def run_axis_finder(input_str):
 
 demo = gr.Interface(
     run_axis_finder,
-    gr.Textbox(value="""0.7 (44,119) 2 
+    gr.Textbox(value="""0.85 (44,119) 2 
 Road Access[0.1:0.9], Starting Point(y,x), Carbon Prefix[0:7]
 Road Access setups the ratio of accessibility for each divided two parts.
 Starting Point is a tuple of pixel location for the algorithm to start its road construction.

axis_finder.py

@@ -128,6 +128,7 @@ def gradio_inference(access_ratio,start_point,carbon_weight):
     config.A_CARBON_WEIGHT = int(carbon_weight)
     run_axis_finder(0)
     image = cv2.imread(f'outputs/collision_map.bmp')
+    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     return image
 
 if __name__ == "__main__":