Update app.py

app.py

@@ -8,7 +8,6 @@ from utils import *
 
 # Initialize labels and model
 labels = gen_labels()
-st.info(print(labels))
 model = model_arc()  # Assuming this function initializes and returns a trained model
 
 # Streamlit UI
@@ -31,11 +30,11 @@ opt = st.selectbox("How do you want to upload the image for classification?",
 image = None
 if opt == 'Upload image from device':
     file = st.file_uploader('Select', type=['jpg', 'png', 'jpeg'])
-    st.info(print(file))
+    print(file)
     if file:
         try:
             image = Image.open(io.BytesIO(file.read())).resize((256, 256), Image.LANCZOS)
-            st.info(print(image))
+            print(image)
         except Exception as e:
             st.error(f"Error reading the file: {e}")
 
@@ -60,9 +59,9 @@ try:
         #model.load_weights("classify_model.h5")
 
         
-        st.info(print(model))
+        print(model)
         prediction = model.predict(img[np.newaxis, ...])
-        st.info(print(prediction))
+        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)