Update README.md

README.md

@@ -96,13 +96,13 @@ NOTE: This pipline has just the overlook and the basic parameters any additional
    # Load ResNet50 model
    model = ResNet50(weights="imagenet", include_top=False, pooling="avg")
 
+   from PIL import Image
    def extract_features(image_path):
-       from PIL import Image
-       image = Image.open(image_path).convert("RGB")
        image = image.resize((224, 224))
-       image_array = np.array(image).astype("float32")
+       image_array = np.array(image)
+       image_array = np.expand_dims(image_array, axis=0).astype("float32")
        image_array = preprocess_input(image_array)
-       image_array = np.expand_dims(image_array, axis=0)
+       
        features = model.predict(image_array)
        return features.flatten()
    ```