Update app.py

app.py

@@ -7,7 +7,7 @@ import io
 from utils import *
 
 # Initialize labels and model
-labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']#gen_labels()
+labels = gen_labels()
 model = model_arc()  # Assuming this function initializes and returns a trained model
 
 # Streamlit UI
@@ -55,6 +55,7 @@ try:
         #model.load_weights("classify_model.h5")
 
         prediction = model.predict(img[np.newaxis, ...])
+        st.info(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)