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

minor changes

Browse files
Files changed (1) hide show
  1. model/pothole_model.py +2 -2
model/pothole_model.py CHANGED
@@ -28,6 +28,7 @@ from PIL import Image, ImageOps
28
  import numpy as np
29
  import requests
30
  from io import BytesIO
 
31
 
32
 
33
 
@@ -36,8 +37,7 @@ def load_image_model(image):
36
  np.set_printoptions(suppress=True)
37
 
38
  # Load the model from the URL
39
- model_url = "keras_model.h5"
40
- model = tf.keras.models.load_model(model_url, compile=False)
41
 
42
  # Load the labels
43
  class_names = open("labels.txt", "r").readlines()
 
28
  import numpy as np
29
  import requests
30
  from io import BytesIO
31
+ from keras.models import load_model
32
 
33
 
34
 
 
37
  np.set_printoptions(suppress=True)
38
 
39
  # Load the model from the URL
40
+ model = load_model("keras_model.h5", compile=False)
 
41
 
42
  # Load the labels
43
  class_names = open("labels.txt", "r").readlines()