Update app.py

app.py

@@ -120,7 +120,7 @@ if st.session_state.image is not None:
     _, col_1, col_2, _ = st.columns([1, 4, 4, 1], gap="medium")
     with col_1:
         st.write('small image 1 here')
-        st.image(pred)
+        # st.image(pred)
     with col_2:
         st.image(st.session_state.image, use_column_width=True)