diff --git a/config.pbtxt b/config.pbtxt
index 9784503..dcaec4b 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -1,21 +1,20 @@
-# Triton backend to use
-name: "resnet50"
 backend: "pytorch"
 platform: "pytorch_libtorch"
-max_batch_size: 0
+
+max_batch_size: 16
 
 input [
+  {
+    name: "input__0"
+    data_type: TYPE_FP32
+    dims: [ 3, 224, 224 ]
+  }
 ]
 
 output [
-]
-
-
-
-
-instance_group [
   {
-    kind: KIND_AUTO
-    count: 1
+    name: "output__0"
+    data_type: TYPE_FP32
+    dims: [ 1000 ]
   }
 ]