Update app.py

app.py

@@ -1,12 +1,9 @@
 import gradio as gr
-from gradio import inputs
 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
 
-
-# Load the model
 model = load_model('dementia.h5')
 
 def classify_image(inp):
@@ -15,7 +12,7 @@ def classify_image(inp):
     prediction = model.predict(inp).flatten()
     return {"Dementia not Detected": float(prediction[0]), "Dementia Detected": float(prediction[1])}
 
-image = gr.inputs.Image(shape=(224, 224))
+image = gr.Image(shape=(224, 224))  # Modify this line
 
 interface = gr.Interface(
     fn=classify_image,
@@ -24,4 +21,4 @@ interface = gr.Interface(
     title="DementAI",
 )
 
-interface.launch()
+interface.launch()