Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -13,7 +13,6 @@ from my_model.gen_utilities import free_gpu_resources
 from my_model.KBVQA import KBVQA, prepare_kbvqa_model
 from my_model.utilities.state_manager import StateManager
 
-global state_manager
 state_manager = StateManager()
 
 def answer_question(caption, detected_objects_str, question, model):
@@ -96,9 +95,10 @@ def process_new_image(image_key, image, kbvqa):
 
 def run_inference():
 
-    
     st.title("Run Inference")
 
+    state_manager.initialize_state()
+
     st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
     st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
     default_confidence = 0.2 if st.session_state['detection_model'] == "yolov5" else 0.4