Spaces:
Sleeping
Sleeping
Soham Chandratre
commited on
Commit
·
85898a0
1
Parent(s):
b3aaca0
minor changes
Browse files- 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 |
-
|
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()
|