Update app.py

app.py

@@ -61,12 +61,12 @@ def predict_age(image):
         age = output.item()  # Convert to a single scalar
     return f"Predicted Age: {age:.2f}"
 
-import gradio as gr
 
-# Replace gr.inputs.Image with just gr.Image
+
+# Update the gr.Image initialization
 iface = gr.Interface(
     fn=predict_age,
-    inputs=gr.Image(shape=(128, 128), image_mode='RGB', source='upload'),
+    inputs=gr.Image(image_size=(128, 128), image_mode='RGB', source='upload'),
     outputs="text",
     title="Age Prediction Model",
     description="Upload an image to predict the age.",
@@ -75,3 +75,4 @@ iface = gr.Interface(
 
 iface.launch()
 
+