Apex-X commited on
Commit
a12dbb7
·
verified ·
1 Parent(s): 28f1a2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -15,13 +15,8 @@ num_classes = len(classes)
15
 
16
  # Load the model
17
 
18
- from keras.models import load_model
19
 
20
- model_path = 'sketch_recognition_model_cnn.h5' # Ensure this path is correct
21
- try:
22
- model = load_model(model_path)
23
- except Exception as e:
24
- raise RuntimeError(f"Failed to load model from {model_path}: {e}")
25
 
26
  # Predict function for the interface
27
  def predict_fn(image):
 
15
 
16
  # Load the model
17
 
 
18
 
19
+ model = load_model('sketch_recogination_model_cnn.h5')
 
 
 
 
20
 
21
  # Predict function for the interface
22
  def predict_fn(image):