Update app.py

app.py

@@ -1,9 +1,10 @@
 import gradio as gr
 from tensorflow.keras.models import load_model
 from tensorflow.keras.preprocessing import image
-import numpy as np
 from keras.applications.imagenet_utils import preprocess_input
+import numpy as np
 
+# Load the model
 model = load_model('dementia.h5')
 
 def classify_image(inp):
@@ -12,13 +13,14 @@ def classify_image(inp):
     prediction = model.predict(inp).flatten()
     return {"Dementia not Detected": float(prediction[0]), "Dementia Detected": float(prediction[1])}
 
-image = gr.Image(shape=(224, 224))  # Modify this line
+# Use 'image_mode' to specify the shape
+image_input = gr.Image(image_mode="RGB", source="upload", shape=(224, 224))
 
 interface = gr.Interface(
     fn=classify_image,
-    inputs=image,
+    inputs=image_input,
     outputs="label",
     title="DementAI",
 )
 
-interface.launch()
+interface.launch()