Update app.py

app.py

@@ -8,7 +8,7 @@ from utils import preprocess_image
 
 # Initialize labels and model
 labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
-model = torch.load('classify_model.pth')
+model = torch.load('classify_model.h5')
 model.eval()
 
 # Customized Streamlit layout