supib4132 commited on
Commit
2800504
·
verified ·
1 Parent(s): 350853d

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -44,7 +44,7 @@ def extract_image_features(image):
44
  # except Exception as e:
45
  # print(f"Error extracting features: {e}")
46
  # return None
47
- try:
48
  # Convert NumPy array to PIL if needed
49
  if isinstance(image, np.ndarray):
50
  image = Image.fromarray(image.astype("uint8")).convert("RGB")
 
44
  # except Exception as e:
45
  # print(f"Error extracting features: {e}")
46
  # return None
47
+ try:
48
  # Convert NumPy array to PIL if needed
49
  if isinstance(image, np.ndarray):
50
  image = Image.fromarray(image.astype("uint8")).convert("RGB")