Update app.py

app.py

@@ -14,6 +14,9 @@ labels = {name: index for index, name in enumerate(classes)}
 num_classes = len(classes)
 
 # Load the model
+
+from keras.models import load_model
+
 model_path = 'sketch_recognition_model_cnn.h5'  # Ensure this path is correct
 try:
     model = load_model(model_path)