Update model path in app.py to use a relative directory
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ img_size = 28
|
|
| 26 |
labels = ["zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
|
| 27 |
|
| 28 |
# load model (trained on MNIST dataset)
|
| 29 |
-
model = tf.keras.models.load_model("
|
| 30 |
|
| 31 |
# prediction function for sketch recognition
|
| 32 |
def predict(img):
|
|
|
|
| 26 |
labels = ["zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
|
| 27 |
|
| 28 |
# load model (trained on MNIST dataset)
|
| 29 |
+
model = tf.keras.models.load_model("./sketch_recognition_numbers_model.h5")
|
| 30 |
|
| 31 |
# prediction function for sketch recognition
|
| 32 |
def predict(img):
|