Update app.py

app.py

@@ -60,14 +60,14 @@ def preprocess_image(image):
     
     image = cv2.resize(image, (224, 224))
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
+    # Convert to PIL and extract
+    image_pil = Image.fromarray(image)
+    inputs = feature_extractor(images=image_pil, return_tensors="pt")
     
-    inputs = feature_extractor(
-        images=[image],
-        return_tensors="pt",
-        padding=True
-    )
     return inputs['pixel_values']
 
+
 def predict_face_shape(image):
     # Force CPU usage on Hugging Face Spaces
     device = torch.device("cpu")