diff --git a/agentic_security/static/base.js b/agentic_security/static/base.js
index b255ff7..98f37ee 100644
--- a/agentic_security/static/base.js
+++ b/agentic_security/static/base.js
@@ -168,8 +168,8 @@ let LLM_CONFIGS = [
   { name: 'Replicate', prompts: 40000 },
   { name: 'Groq', prompts: 40000 },
   { name: 'Together.ai', prompts: 40000 },
-  { name: 'Custom API Image', prompts: 40000, customInstructions: 'Requires api spec' },
-  { name: 'Custom API Files', prompts: 40000, customInstructions: 'Requires api spec' },
+  { name: 'Custom API Image', prompts: 40000, customInstructions: 'Requires api spec', modality: 'Image' },
+  { name: 'Custom API Files', prompts: 40000, customInstructions: 'Requires api spec', modality: 'Files' },
   { name: 'Gemini', prompts: 40000 },
   { name: 'Claude', prompts: 40000 },
   { name: 'Cohere', prompts: 40000 },
diff --git a/agentic_security/static/index.html b/agentic_security/static/index.html
index 21d7686..133895a 100644
--- a/agentic_security/static/index.html
+++ b/agentic_security/static/index.html
@@ -51,7 +51,8 @@
               <div class="text-sm text-gray-400">
                 {{ config.customInstructions || 'Requires API key' }}
               </div>
-              <div class="mt-2 text-dark-accent-green font-semibold">API</div>
+              <div class="mt-2 text-dark-accent-green font-semibold">
+                {{config.modality || 'API'}}</div>
             </div>
           </div>
         </section>