Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -48,10 +48,10 @@ class InferenceRunner(StateManager):
 
         # Display and interact with each uploaded/selected image
         for image_key, image_data in self.get_images_data().items():
-            col2.image(image_data['image'], caption=f'Uploaded Image: {image_key[-11:]}', use_column_width=True)
+            self.col2.image(image_data['image'], caption=f'Uploaded Image: {image_key[-11:]}', use_column_width=True)
             if not image_data['analysis_done']:
                 self.col2.text("Cool image, please click 'Analyze Image'..")
-                if col2.button('Analyze Image', key=f'analyze_{image_key}'):
+                if self.col2.button('Analyze Image', key=f'analyze_{image_key}'):
                     caption, detected_objects_str, image_with_boxes = self.analyze_image(image_data['image'], kbvqa)
                     self.update_image_data(image_key, caption, detected_objects_str, True)
 
@@ -59,8 +59,8 @@ class InferenceRunner(StateManager):
             qa_history = image_data.get('qa_history', [])
 
             if image_data['analysis_done']:
-                question = col2.text_input(f"Ask a question about this image ({image_key[-11:]}):", key=f'question_{image_key}')
-                if col2.button('Get Answer', key=f'answer_{image_key}'):
+                question = self.col2.text_input(f"Ask a question about this image ({image_key[-11:]}):", key=f'question_{image_key}')
+                if self.col2.button('Get Answer', key=f'answer_{image_key}'):
                     if question not in [q for q, _ in qa_history]:
                         answer = self.answer_question(image_data['caption'], image_data['detected_objects_str'], question, kbvqa)
                         self.add_to_qa_history(image_key, question, answer)
@@ -77,15 +77,15 @@ class InferenceRunner(StateManager):
         self.set_up_widgets()
         st.session_state['settings_changed'] = self.has_state_changed()
         if st.session_state['settings_changed']:
-            col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
+            self.col1.warning("Model settings have changed, please reload the model, this will take a second .. ")
 
         st.session_state.button_label = "Reload Model" if self.is_model_loaded() and self.settings_changed else "Load Model"
         
         if st.session_state.method == "Fine-Tuned Model":
-            if col1.button(st.session_state.button_label):
+            if self.col1.button(st.session_state.button_label):
                 if st.session_state.button_label == "Load Model":
                     if self.is_model_loaded():
-                        col1.text("Model already loaded and no settings were changed:)")
+                        self.col1.text("Model already loaded and no settings were changed:)")
                     else: 
                         self.load_model()
                 else:
@@ -94,4 +94,4 @@ class InferenceRunner(StateManager):
             if self.is_model_loaded() and st.session_state.kbvqa.all_models_loaded:
                 self.image_qa_app(self.get_model())
         else:
-            col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')
+            self.col1.warning(f'Model using {st.session_state.method} is not deployed yet, will be ready later.')