fix hello_world/evaluate.py params input_shape

tensorflow · Sep 25, 2024 · 7291c19 · 7291c19
1 parent 2d939f3
commit 7291c19
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow/lite/micro/examples/hello_world/evaluate.py b/tensorflow/lite/micro/examples/hello_world/evaluate.py
@@ -108,7 +108,8 @@ def get_tflite_prediction(model_path, x_values):
 
   for i, x_value in enumerate(x_values):
     y_predictions[i] = invoke_tflite_interpreter(
-        in_interpreter  tflite_interpreter,
+        input_shape,
+        tflite_interpreter,
         x_value,
         input_details['index'],
         output_details['index'],