Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -182,7 +182,7 @@ class InferenceRunner(StateManager):
         
         self.display_sample_images()
         self.handle_image_upload()
-        self.display_session_state()
+        self.display_session_state(self.col1)
         with self.col2:
             for image_key, image_data in self.get_images_data().items():
                 with st.container():