Update config.pbtxt

config.pbtxt

@@ -1,18 +1,25 @@
 backend: "python"
-max_batch_size: 0
+max_batch_size: 0  # Keep batching disabled as per original config
 
 input [
-{
-  name: "input_jp2_bytes"
-  data_type: TYPE_STRING
-  dims: [ 3 ]
-}
+  {
+    name: "input_jp2_bytes"  # New input name for JP2 bytes
+    data_type: TYPE_STRING  # Use TYPE_STRING for bytes
+    dims: [ 3 ]             # Expecting 3 elements: Red, Green, NIR bytes
+  }
 ]
 
 output [
-{
-  name: "output_mask"
-  data_type: TYPE_UINT8
-  dims: [-1, -1]
-}
-]
+  {
+    name: "output_mask"
+    data_type: TYPE_UINT8
+    dims: [-1, -1]     # Variable height, width
+  }
+]
+
+# Optional: Specify instance_group if running on GPU
+# instance_group [
+#   {
+#     kind: KIND_GPU
+#   }
+# ]