Soham Chandratre commited on
Commit
3c93956
·
1 Parent(s): 85898a0

minor changes

Browse files
Files changed (1) hide show
  1. model/pothole_model.py +1 -1
model/pothole_model.py CHANGED
@@ -48,7 +48,7 @@ def load_image_model(image):
48
  data = np.ndarray(shape=(1, 224, 224, 3), dtype=np.float32)
49
 
50
  # Replace this with the path to your image
51
- image = Image.open(image).convert("RGB")
52
 
53
  # resizing the image to be at least 224x224 and then cropping from the center
54
  size = (224, 224)
 
48
  data = np.ndarray(shape=(1, 224, 224, 3), dtype=np.float32)
49
 
50
  # Replace this with the path to your image
51
+ image = Image.open(BytesIO(image)).convert("RGB")
52
 
53
  # resizing the image to be at least 224x224 and then cropping from the center
54
  size = (224, 224)