m7mdal7aj commited on
Commit
ef7f9be
1 Parent(s): c0e58be

Update my_model/KBVQA.py

Browse files
Files changed (1) hide show
  1. my_model/KBVQA.py +1 -0
my_model/KBVQA.py CHANGED
@@ -9,6 +9,7 @@ from my_model.utilities.gen_utilities import free_gpu_resources
9
  from my_model.captioner.image_captioning import ImageCaptioningModel
10
  from my_model.object_detection import ObjectDetector
11
  import my_model.config.kbvqa_config as config
 
12
 
13
 
14
  class KBVQA(StateManager):
 
9
  from my_model.captioner.image_captioning import ImageCaptioningModel
10
  from my_model.object_detection import ObjectDetector
11
  import my_model.config.kbvqa_config as config
12
+ from my_model.state_manager import StateManager
13
 
14
 
15
  class KBVQA(StateManager):