Update age_classification.py

age_classification.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from transformers import AutoImageProcessor
 from transformers import SiglipForImageClassification
 from transformers.image_utils import load_image
@@ -10,6 +11,7 @@ model_name = "prithivMLmods/Age-Classification-SigLIP2"
 model = SiglipForImageClassification.from_pretrained(model_name)
 processor = AutoImageProcessor.from_pretrained(model_name)
 
+@spaces.GPU
 def age_classification(image):
     """Predicts the age group of a person from an image."""
     image = Image.fromarray(image).convert("RGB")