KB-VQA-E

Sleeping

m7mdal7aj commited on Feb 25

Commit

4c2fc41

•

1 Parent(s): 18d1852

Update my_model/tabs/run_inference.py

Files changed (1) hide show

my_model/tabs/run_inference.py CHANGED Viewed

@@ -11,7 +11,7 @@ from my_model.object_detection import detect_and_draw_objects
 from my_model.captioner.image_captioning import get_caption
 from my_model.gen_utilities import free_gpu_resources
 from my_model.KBVQA import KBVQA, prepare_kbvqa_model
-from my_model.utilities.st_utils import UIManager, StateManager
 state_manager = StateManager()

 from my_model.captioner.image_captioning import get_caption
 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()