Fix image path in app.py for class image and model loading
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ title = "Welcome on your first sketch recognition app!"
|
|
9 |
# app description
|
10 |
head = (
|
11 |
"<center>"
|
12 |
-
"<img src='
|
13 |
"The robot was trained to classify numbers (from 0 to 9). To test it, write your number in the space provided."
|
14 |
"</center>"
|
15 |
)
|
@@ -24,7 +24,7 @@ img_size = 28
|
|
24 |
labels = ["zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
|
25 |
|
26 |
# load model (trained on MNIST dataset)
|
27 |
-
model = tf.keras.models.load_model("
|
28 |
|
29 |
# prediction function for sketch recognition
|
30 |
def predict(img):
|
|
|
9 |
# app description
|
10 |
head = (
|
11 |
"<center>"
|
12 |
+
"<img src='./mnist-classes.png' width=400>"
|
13 |
"The robot was trained to classify numbers (from 0 to 9). To test it, write your number in the space provided."
|
14 |
"</center>"
|
15 |
)
|
|
|
24 |
labels = ["zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
|
25 |
|
26 |
# load model (trained on MNIST dataset)
|
27 |
+
model = tf.keras.models.load_model("./sketch_recognition_numbers_model.h5")
|
28 |
|
29 |
# prediction function for sketch recognition
|
30 |
def predict(img):
|