Apex-X commited on
Commit
1c4d0c8
·
verified ·
1 Parent(s): efcc78a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -14,8 +14,9 @@ classes = [
14
  labels = {name: index for index, name in enumerate(classes)}
15
  num_classes = len(classes)
16
 
17
- # Load the model
18
- model = load_model('sketch_recognition_model_cnn.h5')
 
19
 
20
  # Predict function for interface
21
  def predict_fn(image):
 
14
  labels = {name: index for index, name in enumerate(classes)}
15
  num_classes = len(classes)
16
 
17
+ # Load the model - update this path to the actual location of your model file
18
+ model_path = '/path/to/your/model/sketch_recognition_model_cnn.h5'
19
+ model = load_model(model_path)
20
 
21
  # Predict function for interface
22
  def predict_fn(image):