minor changes

model/pothole_model.py

@@ -29,33 +29,44 @@ import numpy as np
 import requests
 from io import BytesIO
 
+def load_image_model(image):
+   import tensorflow as tf
+from PIL import Image, ImageOps
+import numpy as np
+
 def load_image_model(image):
     # Disable scientific notation for clarity
     np.set_printoptions(suppress=True)
 
     # Load the model from the URL
-    model_url = "https://huggingface.co/spaces/Soham0708/pothole_detect/resolve/main/keras_model.h5"
-    model_path = tf.keras.utils.get_file("keras_model.h5", model_url)
-
-    # Load the model
-    model = tf.keras.models.load_model(model_path)
+    model_url = "https://huggingface.co/spaces/Soham0708/pothole_detect/blob/main/keras_model.h5"
+    model = tf.keras.models.load_model(model_url, compile=False)
 
     # Load the labels
     class_names = open("labels.txt", "r").readlines()
 
     # Create the array of the right shape to feed into the keras model
+    # The 'length' or number of images you can put into the array is
+    # determined by the first position in the shape tuple, in this case 1
     data = np.ndarray(shape=(1, 224, 224, 3), dtype=np.float32)
 
     # Replace this with the path to your image
     image = Image.open(image).convert("RGB")
 
-    # Resize and preprocess the image
-    image = ImageOps.fit(image, (224, 224), Image.ANTIALIAS)
+    # resizing the image to be at least 224x224 and then cropping from the center
+    size = (224, 224)
+    image = ImageOps.fit(image, size, Image.Resampling.LANCZOS)
+
+    # turn the image into a numpy array
     image_array = np.asarray(image)
+
+    # Normalize the image
     normalized_image_array = (image_array.astype(np.float32) / 127.5) - 1
+
+    # Load the image into the array
     data[0] = normalized_image_array
 
-    # Make prediction
+    # Predicts the model
     prediction = model.predict(data)
     index = np.argmax(prediction)
     class_name = class_names[index]