Spaces:
Sleeping
Sleeping
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 |
|