Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -58,6 +58,7 @@ class InferenceRunner(StateManager):
         with self.col2:
             for image_key, image_data in self.get_images_data().items():
                 with st.container():
+                    self.display_session_state()
                     nested_col21, nested_col22 = st.columns([0.65, 0.35])
                     image_for_display = self.resize_image(image_data['image'], 600)
                     nested_col21.image(image_for_display, caption=f'Uploaded Image: {image_key[-11:]}')