Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -42,8 +42,15 @@ class InferenceRunner(StateManager):
                 if st.button(f'Select Sample Image {idx + 1}', key=f'sample_{idx}'):
                     self.process_new_image(sample_image_path, image, kbvqa)
 
+                    
+        st.write(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)
 
@@ -156,6 +163,7 @@ class InferenceRunner(StateManager):
         
         if self.is_model_loaded():
             st.session_state['time_taken_to_load_model'] = time.time()-t1
+            st.write(f'time is {time.time()-t1}')
             free_gpu_resources()
             st.session_state['loading_in_progress'] = False
             self.image_qa_app(self.get_model())