Shortlisted tax for demo

app.py

@@ -289,7 +289,7 @@ with gr.Blocks(title="Search-TTA (Simplified)", theme=gr.themes.Base()) as demo:
 
     # EXAMPLES – copied from original demo (satellite, ground, taxonomy only)
     with gr.Row():
-        gr.Markdown("### In-Domain Taxonomy")
+        gr.Markdown("### Taxonomy")
     with gr.Row():
         gr.Examples(
             examples=[
@@ -299,14 +299,9 @@ with gr.Blocks(title="Search-TTA (Simplified)", theme=gr.themes.Base()) as demo:
                     "Animalia Chordata Aves Charadriiformes Laridae Larus marinus",
                 ],
                 [
-                    "examples/Animalia_Chordata_Mammalia_Rodentia_Caviidae_Hydrochoerus_hydrochaeris/28871_-12.80255_-69.29999.jpg",
-                    "examples/Animalia_Chordata_Mammalia_Rodentia_Caviidae_Hydrochoerus_hydrochaeris/1b8064f8-7deb-4b30-98cd-69da98ba6a3d.jpg",
-                    "Animalia Chordata Mammalia Rodentia Caviidae Hydrochoerus hydrochaeris",
-                ],
-                [
-                    "examples/Animalia_Arthropoda_Malacostraca_Decapoda_Ocypodidae_Ocypode_quadrata/277303_38.72364_-75.07749.jpg",
-                    "examples/Animalia_Arthropoda_Malacostraca_Decapoda_Ocypodidae_Ocypode_quadrata/0b9cc264-a2ba-44bd-8e41-0d01a6edd1e8.jpg",
-                    "Animalia Arthropoda Malacostraca Decapoda Ocypodidae Ocypode quadrata",
+                    "examples/Animalia_Chordata_Mammalia_Carnivora_Ursidae_Ursus_americanus/yosemite_v3_resized.png",
+                    "examples/Animalia_Chordata_Mammalia_Carnivora_Ursidae_Ursus_americanus/248820933.jpeg",
+                    "Animalia Chordata Mammalia Carnivora Ursidae Ursus americanus",
                 ],
                 [
                     "examples/Animalia_Chordata_Mammalia_Rodentia_Sciuridae_Marmota_marmota/388246_45.49036_7.14796.jpg",
@@ -325,38 +320,6 @@ with gr.Blocks(title="Search-TTA (Simplified)", theme=gr.themes.Base()) as demo:
             cache_examples=False,
         )
 
-    with gr.Row():
-        gr.Markdown("### Out-Domain Taxonomy")
-    with gr.Row():
-        gr.Examples(
-            examples=[
-                [
-                    "examples/Animalia_Chordata_Mammalia_Carnivora_Phocidae_Mirounga_angustirostris/27423_35.64005_-121.17595.jpg",
-                    "examples/Animalia_Chordata_Mammalia_Carnivora_Phocidae_Mirounga_angustirostris/3aac526d-c921-452a-af6a-cb4f2f52e2c4.jpg",
-                    "Animalia Chordata Mammalia Carnivora Phocidae Mirounga angustirostris",
-                ],
-                [
-                    "examples/Animalia_Chordata_Mammalia_Carnivora_Canidae_Canis_aureus/1528408_13.00422_80.23033.jpg",
-                    "examples/Animalia_Chordata_Mammalia_Carnivora_Canidae_Canis_aureus/37faabd2-a613-4461-b27e-82fe5955ecaf.jpg",
-                    "Animalia Chordata Mammalia Carnivora Canidae Canis aureus",
-                ],
-                [
-                    "examples/Animalia_Chordata_Mammalia_Carnivora_Ursidae_Ursus_americanus/yosemite_v3_resized.png",
-                    "examples/Animalia_Chordata_Mammalia_Carnivora_Ursidae_Ursus_americanus/248820933.jpeg",
-                    "Animalia Chordata Mammalia Carnivora Ursidae Ursus americanus",
-                ],
-                [
-                    "examples/Animalia_Chordata_Mammalia_Carnivora_Canidae_Urocyon_littoralis/304160_34.0144_-119.54417.jpg",
-                    "examples/Animalia_Chordata_Mammalia_Carnivora_Canidae_Urocyon_littoralis/0cbdfbf2-6cfe-4d61-9602-c949f24d0293.jpg",
-                    "Animalia Chordata Mammalia Carnivora Canidae Urocyon littoralis",
-                ],
-            ],
-            inputs=[sat_input, ground_input, taxonomy_input],
-            outputs=[run_tta_btn, run_no_tta_btn, display_img_tta],
-            fn=process_with_tta,
-            cache_examples=False,
-        )
-
 
     run_tta_btn.click(
         fn=process_with_tta,
@@ -383,8 +346,8 @@ if __name__ == "__main__":
     from app_multimodal_inference import demo as multimodal_demo
 
     combined_demo = gr.TabbedInterface(
-        [multimodal_demo, demo],
-        ["Multimodal Inference", "Search-TTA"]
+        [demo, multimodal_demo],
+        ["Search-TTA", "Multimodal Inference"]
     )
     combined_demo.queue(max_size=15)
     combined_demo.launch(share=True)

test_parameter.py

@@ -45,7 +45,7 @@ VIZ_GRAPH_EDGES = False  # do you want to visualize the graph edges
 # MODEL_NAME = "vlm_search_20x20_200125_256steps_CORRECT_REWARDS.pth" # checkpoint.pth
 MODEL_NAME = "STAGE1_vlm_search_24x24_040425_no_tgt_rewards_iNAT_DS_16k.pth" 
 
-NUM_EPS_STEPS = getenv("NUM_EPS_STEPS", default=384, cast_type=int)
+NUM_EPS_STEPS = getenv("NUM_EPS_STEPS", default=256, cast_type=int)
 TERMINATE_ON_TGTS_FOUND = False  # Whether to terminate episode when all targets found
 FORCE_LOGGING_DONE_TGTS_FOUND = True  # Whether to force csv logging when all targets found
 FIX_START_POSITION = getenv("FIX_START_POSITION", default=True, cast_type=bool)  # Whether to fix the starting position of the robots (middle index)