move model loading to first selection of image.

app.py

@@ -198,6 +198,7 @@ PART_IMAGES_DICT = {file_name: load_part_images(file_name) for file_name in IMAG
 
 # --- Gradio Functions ---
 def update_selected_image(event: gr.SelectData):
+    initialize_model()
     image_height = 400
     index = event.index
 
@@ -382,7 +383,6 @@ custom_css = """
 
 # Define the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(), css=custom_css, title="PEEB") as demo:
-    initialize_model()
     current_image = gr.State("")
     current_predicted_class = gr.State("")
     gt_class = gr.State("")