Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -13,7 +13,7 @@ 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
 
-state_manager = StateManager()
+
 
 def answer_question(caption, detected_objects_str, question, model):
     free_gpu_resources()
@@ -94,6 +94,9 @@ def process_new_image(image_key, image, kbvqa):
         }
 
 def run_inference():
+    global state_manager
+    state_manager = StateManager()
+    
     st.title("Run Inference")
 
     st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')