Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -9,6 +9,7 @@ from my_model.utilities.gen_utilities import free_gpu_resources
 from my_model.captioner.image_captioning import ImageCaptioningModel
 from my_model.object_detection import ObjectDetector
 import my_model.config.kbvqa_config as config
+from my_model.state_manager import StateManager
 
 
 class KBVQA(StateManager):