diff --git a/python/pokedex_rpi.py b/python/pokedex_rpi.py
index 91020193dd81bb1ec87e6655e29a0af9935dedf3..a14d3428e7f881e62615346b898a75f44cfa2e61 100644
--- a/python/pokedex_rpi.py
+++ b/python/pokedex_rpi.py
@@ -25,21 +25,23 @@ else:
 with open(json_path, "r") as f:
     class_names = json.load(f)
 
-# --- Run inference ---
+# --- Inference ---
 with Hailo(hef_path) as hailo:
-    # Get model input shape (e.g., 224x224x3 or 256x256x3)
     model_h, model_w, _ = hailo.get_input_shape()
 
-    # Setup and start the camera
+    # Initialize camera and configure MAIN output only (no preview)
     picam2 = Picamera2()
-    main = {'size': (model_w, model_h), 'format': 'RGB888'}
-    config = picam2.create_preview_configuration(main)
-    picam2.start(config)
+    config = picam2.create_still_configuration(
+        main={"size": (model_w, model_h), "format": "RGB888"},
+        display=None  # Disable preview stream to avoid tiny resolution
+    )
+    picam2.configure(config)
+    picam2.start()
 
     print("-- Capturing image...")
     frame = picam2.capture_array()
 
-    # Optionally display the captured image
+    # Optional: Show the captured image
     try:
         cv2.imshow("Captured Image", frame)
         print("-- Press any key to continue...")
@@ -51,11 +53,9 @@ with Hailo(hef_path) as hailo:
         cv2.imwrite(output_path, frame)
         os.system(f"feh --fullscreen {output_path}")
 
-    # Run inference
     print("-- Running inference...")
     inference_results = hailo.run(frame)
 
-    # Postprocess: find predicted class
     predicted_idx = int(np.argmax(inference_results))
     predicted_name = class_names[predicted_idx]
-    print(f"-- Predicted Pokémon: {predicted_name}")
+    print(f"-- Predicted Pokémon: {predicted_name}")
\ No newline at end of file