Update app.py

app.py

@@ -2,11 +2,15 @@ from transformers import pipeline
 import gradio as gr
 from PIL import Image
 
+# Initialize the image classification pipeline
+pipe = pipeline("image-classification", model="prithivMLmods/Age-Classification-SigLIP2")
+
 # Prediction function
 def predict(input_img):
     # Get the predictions from the pipeline
     predictions = pipe(input_img)
     
+    # Format the result correctly for Gradio output
     result = {p["label"]: p["score"] for p in predictions}
     
     # Return the image and the top predictions as a string
@@ -27,7 +31,3 @@ gradio_app = gr.Interface(
 
 # Launch the app
 gradio_app.launch()
-
-
-if _name=="main_":
-  gradio_app.launch()