diff --git a/assets/large_language_models/rag/components/image_embed_index/spec.yaml b/assets/large_language_models/rag/components/image_embed_index/spec.yaml
index cabc16b6cc..e1f6a6293d 100644
--- a/assets/large_language_models/rag/components/image_embed_index/spec.yaml
+++ b/assets/large_language_models/rag/components/image_embed_index/spec.yaml
@@ -4,7 +4,7 @@ type: command
 tags:
     Preview: ""
 
-version: 0.0.2
+version: 0.0.3
 name: llm_rag_image_embed_index
 display_name: LLM - Embedding images with Florence 
 is_deterministic: true
@@ -16,9 +16,10 @@ outputs:
     description: Where to store the output MLIndex.
 inputs:
   input_path:
-    type: string
+    type: uri_folder
+    mode: rw_mount
     optional: False
-    description: input dataset of images and json metadata.
+    description: The input uri_folder containing a set of images and json metadata.
   embedding_connection_id:
     type: string
     optional: False