diff --git a/python/pokedex_rpi.py b/python/pokedex_rpi.py
index 34dca023eeeb731aa9817d3c5033d6087edc1c73..01101404a342540e398fe1959a400730b4ec92f7 100644
--- a/python/pokedex_rpi.py
+++ b/python/pokedex_rpi.py
@@ -29,7 +29,11 @@ with open(json_path, "r") as f:
 
 # --- Hailo Inference with PiCamera2 ---
 with Hailo(hef_path) as hailo:
-    print(f"-- Model input shape: {input_shape}")
+    model_h, model_w, model_c = hailo.get_input_shape()
+    print(f"-- Model input shape from Hailo: {(model_h, model_w, model_c)}")
+    print(f"-- Input format: {hailo.get_input_format()}")
+    print(f"-- Expected input buffer size: {hailo.get_input_frame_size()}")
+
 
     picam2 = Picamera2()
     config = picam2.create_still_configuration(