Spaces:
Runtime error
Runtime error
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()
|