Update new_app.py

new_app.py

@@ -124,7 +124,12 @@ with col2:
             st.error(f"Error fetching image: {e}")
 
     if state.img is not None:
-        st.image(state.img, width=640) #use_container_width = True
+        st.markdown(
+            "<div style='display:flex; justify-content:center;'>",
+            unsafe_allow_html=True
+        )
+        st.image(state.img, width=640)
+        st.markdown("</div>", unsafe_allow_html=True)
 
 with col1:
     st.subheader("Prediction")