Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from tensorflow.keras.models import load_model
+import tensorflow as tf
 from tensorflow.keras.preprocessing import image
 import numpy as np
 
-# Load your trained model
-model = load_model("waste_sort_model.h5")
+# Load your trained model (SavedModel format)
+model = tf.keras.models.load_model("waste_sort_model")  # Loading from SavedModel directory
 class_names = ["Non-Recyclable", "Recyclable"]
 
 def classify_image(img):
@@ -29,3 +29,4 @@ interface = gr.Interface(
 # Launch the Gradio app
 if __name__ == "__main__":
     interface.launch()
+