Update app.py

app.py

@@ -4,8 +4,8 @@ from tensorflow.keras.preprocessing.image import img_to_array
 import numpy as np
 from PIL import Image
 
-# Load model
-model = load_model("waste_classification(Mobilenetv2).h5")
+# Load the trained model
+model = load_model("waste_classifier_model.h5")
 class_names = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
 
 # Prediction function
@@ -20,15 +20,12 @@ def predict_from_webcam(image):
 
     return f"{predicted_class} ({confidence*100:.1f}%)"
 
-# Use gr.Camera instead of gr.Image
+# Gradio interface with webcam input
 gr.Interface(
     fn=predict_from_webcam,
-    inputs=gr.Image(shape=(224, 224), tool=None, type="pil", label="Capture Image from Webcam"),
+    inputs=gr.Image(source="webcam", tool=None, type="pil"),
     outputs="text",
     live=True,
     title="Live Waste Classifier",
-    description="Click webcam icon and show a waste image to classify."
-)
-
-
-interface.launch()
+    description="Show waste to your webcam. This app predicts if it's paper, glass, plastic, etc."
+).launch()