Update app.py

app.py

@@ -55,6 +55,11 @@ try:
         #model.load_weights("classify_model.h5")
 
         prediction = model.predict(img[np.newaxis, ...])
+
+        print(model)
+        print("------------------------------------")
+        print(prediction)
+        
         st.info('Hey! The uploaded image has been classified as " {} waste " '.format(labels[np.argmax(prediction[0], axis=-1)]))
 except Exception as e:
   st.info(e)