make edits from QA

viamrobotics · Nov 13, 2024 · b6c2e66 · b6c2e66
1 parent 0195d06
commit b6c2e66
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/viam/services/mlmodel/mlmodel.py b/src/viam/services/mlmodel/mlmodel.py
@@ -41,8 +41,12 @@ async def infer(
 
             my_mlmodel = MLModelClient.from_robot(robot=machine, name="my_mlmodel_service")
 
-            nd_array = np.array([1, 2, 3], dtype=np.float64)
-            input_tensors = {"0": nd_array}
+            image_data = np.zeros((1, 384, 384, 3), dtype=np.uint8)
+
+            # Create the input tensors dictionary
+            input_tensors = {
+                "image": image_data
+            }
 
             output_tensors = await my_mlmodel.infer(input_tensors)