disable upload for 3D heads

app.py

@@ -71,6 +71,12 @@ HEADS_REQUIRING_MASK: set[str] = {
     "subarticular_stenosis",
 }
 
+HEADS_3D = {
+    "oasis",
+    "luna16-3D",
+    "kneeMRI",
+}
+
 
 DATASET_OPTIONS: Dict[str, str] = {
     "anatomy-ct": "Anatomy CT (test)",
@@ -295,6 +301,13 @@ def update_upload_component_state(head: str) -> Tuple[Dict[str, Any], Dict[str,
         )
         upload_update = gr.update(interactive=False)
         return info_update, upload_update
+    elif head in HEADS_3D:
+        info_update = gr.update(
+            value="⚠️ Custom image upload is disabled for this task because it requires a 3D image.",
+            visible=True,
+        )
+        upload_update = gr.update(interactive=False)
+        return info_update, upload_update
 
     info_update = gr.update(visible=False)
     upload_update = gr.update(interactive=True)