Update app.py

app.py

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