| # [DEBUG] | |
| from osgeo import gdal | |
| import os | |
| from config import OUTPUT_DIR | |
| import gradio as gr | |
| import logging | |
| from inference_tab import get_inference_widgets, run_inference,georefImg | |
| from annotation_tab import get_annotation_widgets | |
| from map_tab import get_map_widgets | |
| # setup logging | |
| logging.basicConfig(level=logging.DEBUG) | |
| with gr.Blocks() as demo: | |
| selected_tile_state=gr.State(value=None) | |
| with gr.Tab("Inference"): | |
| image_input, gcp_input, city_name,user_crs, score_th, hist_th, hist_dic, run_button, output, download_file = get_inference_widgets(run_inference,georefImg, selected_tile_state) | |
| with gr.Tab("Annotation"): | |
| get_annotation_widgets(selected_tile_state) | |
| with gr.Tab("Map"): | |
| get_map_widgets(city_name) | |
| # [DEBUG] | |
| demo.launch(inbrowser=True) | |
| # [PROD] | |
| #demo.launch(server_name="0.0.0.0", server_port=7860, inbrowser=False) | |