Update app.py

app.py

@@ -50,8 +50,6 @@ if 'image' in locals():  # Check if image variable exists
             
             # Ensure image shape is correct and add batch dimension
             img_array = preprocess(image)  # This should return an array of shape (1, 256, 256, 3)
-
-            st.info('Worked2')
             
             predicted_class, confidence = classify_garbage(img_array, model)
 
@@ -62,5 +60,4 @@ if 'image' in locals():  # Check if image variable exists
             st.info('Worked4')
             
         except Exception as e:
-            st.info('Error')
             st.error(f"An error occurred: {e}")