Apex-X commited on
Commit
377ea58
·
verified ·
1 Parent(s): 1fe0675

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -15,9 +15,8 @@ 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 = '/home/user/models/sketch_recognition_model_cnn.h5' # Change this
19
- model = load_model(model_path)
20
-
21
  # Predict function for interface
22
  def predict_fn(image):
23
  """
 
15
  num_classes = len(classes)
16
 
17
  # Load the model - update this path to the actual location of your model file
18
+ from keras.models import load_model
19
+ model = load_model('sketch_recogination_model_cnn.h5')
 
20
  # Predict function for interface
21
  def predict_fn(image):
22
  """