Update src/streamlit_app.py

src/streamlit_app.py

@@ -212,7 +212,7 @@ if st.session_state.current_image is not None:
         if st.button("dog"):
             print("dog pressed")
             st.session_state.label_input="dog"
-
+    print(f"SHAPE:{st.session_state.current_image.shape}")
     prediction = st.session_state.model.predict(np.array([st.session_state.current_image]))[0][0]
 
     st.session_state.current_prediction = 'dog' if prediction > 0.5 else 'cat'