diff --git a/config.pbtxt b/config.pbtxt
index febb81f..3b088f8 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -5,9 +5,25 @@ platform: "tensorflow_savedmodel"
 max_batch_size: 0
 
 input [
+  {
+    name: "test1"
+    data_type: TYPE_UINT8
+    dims: [ 1 ]
+    reshape {
+      shape: [ 1 ]
+    }
+  }
 ]
 
 output [
+  {
+    name: "test2"
+    data_type: TYPE_UINT16
+    dims: [ 2 ]
+    reshape {
+      shape: [ 2 ]
+    }
+  }
 ]