alibayram commited on
Commit
4747b05
·
1 Parent(s): 4c5bd22

Fix image path in app.py for class image and model loading

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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='file/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,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("model/sketch_recognition_numbers_model.h5")
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):