Update app.py

app.py

@@ -46,18 +46,13 @@ if 'image' in locals():  # Check if image variable exists
     if st.button('Predict'):
         try:
             model = model_arc()  # Initialize your model
-            st.info('Worked1')
             
             # Ensure image shape is correct and add batch dimension
             img_array = preprocess(image)  # This should return an array of shape (1, 256, 256, 3)
             
             predicted_class, confidence = classify_garbage(img_array, model)
-
-            st.info('Worked3')
             
             st.info('The uploaded image has been classified as "{}" with {:.2f}% confidence.'.format(predicted_class, confidence))
-
-            st.info('Worked4')
             
         except Exception as e:
             st.error(f"An error occurred: {e}")