Update my_model/tabs/run_inference.py
Browse files
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 |
-
|
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 |
|