diff --git a/config.pbtxt b/config.pbtxt
index c33e1e1..c7115d5 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -4,6 +4,14 @@ platform: "pytorch_libtorch"
 max_batch_size: 0
 
 input [
+  {
+    name: "input1"
+    data_type: TYPE_UINT8
+    dims: [ 1, 16 ]
+    reshape {
+      shape: [ 1, 16 ]
+    }
+  }
 ]
 
 output [