Update app.py

app.py

@@ -12,9 +12,10 @@ import pickle
 import gradio as gr
 
 def load_model():
-    save_path = 'model.pkl'
-    with open(save_path, 'rb') as file:
-        model = pickle.load(file)
+    save_path = 'model_architecture.json'
+    with open(save_path, 'r') as file:
+        model = keras.models.model_from_json(file.read())
+    model.load_weights('model_weights.h5')
     return model
 
 def predict_dementia(images, model):