Spaces:
Sleeping
Sleeping
Soham Chandratre
commited on
Commit
·
b3aaca0
1
Parent(s):
80940eb
minor changes
Browse files- model/pothole_model.py +1 -1
model/pothole_model.py
CHANGED
|
@@ -36,7 +36,7 @@ def load_image_model(image):
|
|
| 36 |
np.set_printoptions(suppress=True)
|
| 37 |
|
| 38 |
# Load the model from the URL
|
| 39 |
-
model_url =
|
| 40 |
model = tf.keras.models.load_model(model_url, compile=False)
|
| 41 |
|
| 42 |
# Load the labels
|
|
|
|
| 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
|