hard target matching removed from check_mask_stats

app.py

@@ -137,6 +137,21 @@ IMAGE_INFERENCE_MODES = [
     "BIOMED SEGMENTATION + DETECTION + RECOGNITION"
 ]
 
+MODALITY_PROMPTS = {
+   "CT-Abdomen": ["postcava", "aorta", "right kidney", "kidney", "left kidney", "duodenum", "pancreas", "liver", "spleen", "stomach", "gallbladder", "left adrenal gland", "adrenal gland", "right adrenal gland", "esophagus"],
+   "CT-Chest": ["nodule", "COVID-19 infection", "tumor"],
+   "MRI-Abdomen": ["aorta", "postcava", "right kidney", "duodenum", "kidney", "left kidney", "liver", "pancreas", "gallbladder", "stomach", "spleen", "left adrenal gland", "adrenal gland", "right adrenal gland", "esophagus"],
+   "MRI-Cardiac": ["left heart ventricle", "myocardium", "right heart ventricle"],
+   "MRI-FLAIR-Brain": ["edema", "tumor core", "whole tumor"],
+   "MRI-T1-Gd-Brain": ["enhancing tumor", "non-enhancing tumor", "tumor core"],
+   "Pathology": ["connective tissue cells", "inflammatory cells", "neoplastic cells", "epithelial cells"],
+   "X-Ray-Chest": ["left lung", "lung", "right lung"],
+   "Ultrasound-Cardiac": ["left heart atrium", "left heart ventricle"],
+   "Endoscopy": ["neoplastic polyp", "polyp", "non-neoplastic polyp"],
+   "Fundus": ["optic cup", "optic disc"],
+   "Dermoscopy": ["lesion", "melanoma"],
+   "OCT": ["edema"] }
+
 
 def on_mode_dropdown_change(selected_mode):
     if selected_mode in IMAGE_INFERENCE_MODES:
@@ -176,6 +191,11 @@ def initialize_model():
 
 model = initialize_model()
 
+def update_example_prompts(modality):
+    if modality in MODALITY_PROMPTS:
+        examples = MODALITY_PROMPTS[modality]
+        return f"Example prompts for {modality}:\n" + ", ".join(examples)
+    return ""
 
 # Utility functions
 @spaces.GPU
@@ -250,7 +270,10 @@ with gr.Blocks() as demo:
                 interactive=False,
                 show_label=True
             )
-    
+        with gr.Accordion("Example Prompts by Modality", open=False):
+            for modality, prompts in MODALITY_PROMPTS.items():
+                prompt_str = ", ".join(prompts)
+                gr.Markdown(f"**{modality}**: {prompt_str}")
     # Add error handling for the submit button
     submit_btn.click(
         fn=process_image,