Fix Gradio compatibility: remove deprecated @gr .cache and add model download from HF Hub

app.py

@@ -3,15 +3,21 @@ import tensorflow as tf
 import numpy as np
 from PIL import Image
 import cv2
+from huggingface_hub import hf_hub_download
 
 # Load the model
-@gr.cache
 def load_model():
     """Load the age group classification model"""
     try:
-        model = tf.keras.models.load_model('resnet50v2_age_classifier_best.h5')
+        # Download model from Hugging Face Hub
+        model_path = hf_hub_download(
+            repo_id="Sharris/age-group-classifier",
+            filename="resnet50v2_age_classifier_best.h5"
+        )
+        model = tf.keras.models.load_model(model_path)
         return model
-    except:
+    except Exception as e:
+        print(f"Error loading model: {e}")
         # Fallback if model file not found
         return None