Update my_model/state_manager.py

my_model/state_manager.py

@@ -246,7 +246,7 @@ class StateManager:
             st.error(f"Error reloading detection model: {e}")
 
     
-    def process_new_image(self, image_key, image, kbvqa):
+    def process_new_image(self, image_key, image):
         """
         Processes a new uploaded image by creating an entry in the `images_data` dictionary in the application session state.
     
@@ -273,7 +273,7 @@ class StateManager:
             }
 
     
-    def analyze_image(self, image, kbvqa):
+    def analyze_image(self, image):
         """
         Analyzes the image using the KBVQA model.
 
@@ -291,8 +291,8 @@ class StateManager:
         """
         img = copy.deepcopy(image)
         st.text("Analyzing the image .. ")
-        caption = kbvqa.get_caption(img)
-        image_with_boxes, detected_objects_str = kbvqa.detect_objects(img)
+        caption = self.session_state.kbvqa.get_caption(img)
+        image_with_boxes, detected_objects_str = self.session_state.kbvqa.detect_objects(img)
         return caption, detected_objects_str, image_with_boxes