Update app.py

app.py

@@ -2,10 +2,13 @@ import gradio as gr
 import tensorflow as tf
 from tensorflow import keras
 
+import huggingface_hub
+from huggingface_hub import from_pretrained_keras
 
-# load pre-trained model
-model_path = "/Users/chaninderrishi/Desktop/ML/projects/waste-sorting/models/prod3"
-pre_trained_model = keras.models.load_model(model_path)
+
+# load custom pre-trained model from HuggingFace models
+model_api_link = 'chaninder/waste-sorting-model'
+pre_trained_model = from_pretrained_keras(model_api_link)
 
 # classification labels
 labels = ['compost', 'e-waste', 'recycle', 'trash']