Update app.py

app.py

@@ -59,12 +59,13 @@ def predict_age(image):
         age = output.item()  # Convert to a single scalar
     return f"Predicted Age: {age:.2f}"
 
-iface = gr.Interface(
-    fn=predict_age,
-    inputs=gr.Image(image_mode='RGB', source='upload'),  # Removed image_size
-    outputs="text",
-    title="Age Prediction Model",
-    description="Upload an image to predict the age.",
-    live=True
-)
+import gradio as gr
+
+gr.Interface(
+    fn=predict_fn,
+    inputs=gr.Image(type="pil", image_mode="RGB"),
+    outputs="number",  # or "text" if your output is text-based
+    title="Age Prediction from Face",
+    description="Upload a face image and get the predicted age."
+).launch()