Update inference.py

inference.py

@@ -31,11 +31,28 @@ def extract_image_features(image):
     Input: PIL Image or image path (str).
     Output: Normalized image embedding (numpy array).
     """
-    try:
-        if isinstance(image, str):
+    # try:
+    #     if isinstance(image, str):
+    #         image = Image.open(image).convert("RGB")
+    #     else:
+    #         image = image.convert("RGB")
+    #     inputs = clip_processor(images=image, return_tensors="pt")
+    #     with torch.no_grad():
+    #         features = clip_model.get_image_features(**inputs)
+    #     features = torch.nn.functional.normalize(features, p=2, dim=-1)
+    #     return features.squeeze(0).cpu().numpy().astype("float32")
+    # except Exception as e:
+    #     print(f"Error extracting features: {e}")
+    #     return None
+     try:
+        # Convert NumPy array to PIL if needed
+        if isinstance(image, np.ndarray):
+            image = Image.fromarray(image.astype("uint8")).convert("RGB")
+        elif isinstance(image, str):
             image = Image.open(image).convert("RGB")
         else:
             image = image.convert("RGB")
+
         inputs = clip_processor(images=image, return_tensors="pt")
         with torch.no_grad():
             features = clip_model.get_image_features(**inputs)