Update pages/2_Player_Comparison_Throw_Image.py

pages/2_Player_Comparison_Throw_Image.py

@@ -47,15 +47,18 @@ def detect_and_predict_face(image_file, model):
 
     expected_features = model.named_steps['scaler'].n_features_in_
 
-    # ✅ CORRECT: width=113, height=109 → 113*109 = 12289
-    resized_face = cv2.resize(face, (113, 109))  # <--- this is THE FIX
+    # Resize to shape close to the expected features
+    resized_face = cv2.resize(face, (113, 109))  # width x height
 
     flattened = resized_face.flatten().reshape(1, -1)
 
-    # Debug output
-    st.write(f"Resized shape: {resized_face.shape}")     # should be (109, 113)
-    st.write(f"Flattened shape: {flattened.shape}")      # should be (1, 12289)
+    # 🔧 Fix: if extra pixels due to rounding, trim them
+    if flattened.shape[1] > expected_features:
+        flattened = flattened[:, :expected_features]
+    elif flattened.shape[1] < expected_features:
+        return None, f"❌ Still too small after resizing. Got {flattened.shape[1]}"
 
+    # Final sanity check
     if flattened.shape[1] != expected_features:
         return None, f"❌ Mismatch in feature count. Got {flattened.shape[1]}, expected {expected_features}"
 
@@ -68,6 +71,8 @@ def detect_and_predict_face(image_file, model):
 
 
 
+
+
 # Get player details from the dataset
 def get_player_details(df, player_name):
     return df[df['Player'] == player_name].iloc[0]