Update app.py

app.py

@@ -62,7 +62,7 @@ def predict_age(image):
 # -------- GRADIO INTERFACE --------
 iface = gr.Interface(
     fn=predict_age,
-    inputs=gr.Image(shape=(128, 128), image_mode='RGB', source='upload'),  # Updated input format
+    inputs=gr.Image(image_size=(128, 128), image_mode='RGB', source='upload'),  # Corrected argument
     outputs="text",
     title="Age Prediction Model",
     description="Upload an image to predict the age.",