minor changes

model/pothole_model.py

@@ -26,6 +26,8 @@
 from keras.models import load_model  # TensorFlow is required for Keras to work
 from PIL import Image, ImageOps  # Install pillow instead of PIL
 import numpy as np
+from io import BytesIO
+import requests
 
 
 def load_image_model(image):
@@ -33,7 +35,10 @@ def load_image_model(image):
     np.set_printoptions(suppress=True)
 
     # Load the model
-    model = load_model("https://huggingface.co/spaces/Soham0708/pothole_detect/blob/main/keras_model.h5")
+    model_url = "https://huggingface.co/spaces/Soham0708/pothole_detect/blob/main/keras_model.h5"
+    response = requests.get(model_url)
+    response.raise_for_status()  # Raise an exception if the download fails
+    model = load_model(BytesIO(response.content))
 
     # Load the labels
     class_names = open("labels.txt", "r").readlines()
@@ -44,7 +49,7 @@ def load_image_model(image):
     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")
+    image = Image.open(BytesIO(image)).convert("RGB")
 
     # resizing the image to be at least 224x224 and then cropping from the center
     size = (224, 224)