Update app.py

app.py

@@ -6,12 +6,16 @@ import gradio as gr
 import numpy as np
 import tensorflow as tf
 from PIL import Image
+from huggingface_hub import hf_hub_download
 
-model = tf.keras.models.load_model(
-    "waste_classifier_final.h5",
-    compile=False
+# Download model from your model repo
+model_path = hf_hub_download(
+    repo_id="dk00069/WasteClassifier01",
+    filename="waste_classifier_final.h5"
 )
 
+model = tf.keras.models.load_model(model_path, compile=False)
+
 labels = [
     "Cardboard",
     "Glass",