Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -90,12 +90,14 @@ class InferenceRunner(StateManager):
                             if st.button('Analyze Image', key=f'analyze_{image_key}', on_click=self.disable_widgets, disabled=self.is_widget_disabled):
                                 
                                 caption, detected_objects_str, image_with_boxes = self.analyze_image(image_data['image'])
+                                st.text("AAAAA")
                                 self.update_image_data(image_key, caption, detected_objects_str, True)
+                                st.text("BBBBB")
                             st.session_state['loading_in_progress'] = False
         
                     # Initialize qa_history for each image
                     qa_history = image_data.get('qa_history', [])
-        
+                    st.text('analysis done ?',image_data['analysis_done'])
                     if image_data['analysis_done']:
                         st.session_state['loading_in_progress'] = False
                         sample_questions = config.SAMPLE_QUESTIONS.get(image_key, [])