Update app.py

app.py

@@ -242,6 +242,7 @@ if st.session_state.screen == 2:
             st.rerun()
         
         # plot predicted image
+        st.markdown(f"<h4 style='text-align: center; color: black;'>Image {str(st.session_state.image_counter + 1)}</h4>", unsafe_allow_html=True)
         im_bgr = st.session_state.results[st.session_state.image_counter].plot()  # BGR-order numpy array
         im_rgb = Image.fromarray(im_bgr[..., ::-1])  # RGB-order PIL image
         st.image(im_rgb)