m7mdal7aj commited on
Commit
e00ca5f
1 Parent(s): a11ed8b

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +3 -3
my_model/tabs/run_inference.py CHANGED
@@ -13,7 +13,8 @@ from my_model.gen_utilities import free_gpu_resources
13
  from my_model.KBVQA import KBVQA, prepare_kbvqa_model
14
  from my_model.utilities.state_manager import StateManager
15
 
16
-
 
17
 
18
  def answer_question(caption, detected_objects_str, question, model):
19
  free_gpu_resources()
@@ -94,8 +95,7 @@ def process_new_image(image_key, image, kbvqa):
94
  }
95
 
96
  def run_inference():
97
- global state_manager
98
- state_manager = StateManager()
99
 
100
  st.title("Run Inference")
101
 
 
13
  from my_model.KBVQA import KBVQA, prepare_kbvqa_model
14
  from my_model.utilities.state_manager import StateManager
15
 
16
+ global state_manager
17
+ state_manager = StateManager()
18
 
19
  def answer_question(caption, detected_objects_str, question, model):
20
  free_gpu_resources()
 
95
  }
96
 
97
  def run_inference():
98
+
 
99
 
100
  st.title("Run Inference")
101