Update app.py

app.py

@@ -59,14 +59,12 @@ def predict_age(image):
         age = output.item()  # Convert to a single scalar
     return f"Predicted Age: {age:.2f}"
 
-# -------- GRADIO INTERFACE --------
 iface = gr.Interface(
     fn=predict_age,
-    inputs=gr.Image(image_size=(128, 128), image_mode='RGB', source='upload'),  # Corrected argument
+    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
 )
 
-iface.launch()