Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -34,7 +34,6 @@ class InferenceRunner(StateManager):
         # Display sample images as clickable thumbnails
         self.col1.write("Choose from sample images:")
         cols = self.col1.columns(len(self.sample_images))
-        st.write('loading in prog?', st.session_state['loading_in_progress'])
         for idx, sample_image_path in enumerate(self.sample_images):
             with cols[idx]:
                 image = Image.open(sample_image_path)
@@ -44,13 +43,11 @@ class InferenceRunner(StateManager):
                     self.process_new_image(sample_image_path, image, kbvqa)
 
                     
-        st.write('loading in prog?',st.session_state['loading_in_progress'])
 
         
         # Image uploader
         uploaded_image = self.col1.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
 
-        st.write(st.session_state['loading_in_progress'])
         
         if uploaded_image is not None:
             self.process_new_image(uploaded_image.name, Image.open(uploaded_image), kbvqa)
@@ -186,5 +183,5 @@ class InferenceRunner(StateManager):
             free_gpu_resources()
             st.session_state['loading_in_progress'] = False
             self.image_qa_app(self.get_model())
-            st.write('load success', 'loading in prog?', st.session_state['loading_in_progress'])
+