diff --git a/config.pbtxt b/config.pbtxt
index 834903e..59f5e51 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -7,9 +7,21 @@ max_batch_size: 0
 default_model_filename: "yolo11n.engine"
 
 input [
+  {
+    name: "images"
+    data_type: TYPE_FP32
+    dims: [1, 3, 640, 640]
+  }
 ]
 
+# Triton should expect to respond with a single string
+# output of variable length named 'text_output'
 output [
+  {
+    name: "output0"
+    data_type: TYPE_FP32
+    dims: [1, 84, 8400]
+  }
 ]