Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -47,9 +47,9 @@ class InferenceRunner(StateManager):
 
         # Display and interact with each uploaded/selected image
         with self.col2:
-            with st.container():
-                nested_col21, nested_col22 = st.columns([0.7, 0.3])
-                for image_key, image_data in self.get_images_data().items():
+            for image_key, image_data in self.get_images_data().items():
+                with st.container():
+                    nested_col21, nested_col22 = st.columns([0.7, 0.3])
                     image_for_display = self.resize_image(image_data['image'], 600)
                     nested_col21.image(image_for_display, caption=f'Uploaded Image: {image_key[-11:]}')
                     if not image_data['analysis_done']: