diff --git a/config.pbtxt b/config.pbtxt
index febb81f..350e9cd 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -5,9 +5,33 @@ platform: "tensorflow_savedmodel"
 max_batch_size: 0
 
 input [
+  {
+    name: "test"
+    data_type: TYPE_UINT32
+    dims: [ 1 ]
+    reshape {
+      shape: [ 1 ]
+    }
+  },
+  {
+    name: "test2"
+    data_type: TYPE_UINT8
+    dims: [ 1 ]
+    reshape {
+      shape: [ 1 ]
+    }
+  }
 ]
 
 output [
+  {
+    name: "testo"
+    data_type: TYPE_UINT8
+    dims: [ 123 ]
+    reshape {
+      shape: [ 123 ]
+    }
+  }
 ]