Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -7,7 +7,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from typing import Tuple, Optional
 from my_model.utilities.gen_utilities import free_gpu_resources
 from my_model.captioner.image_captioning import ImageCaptioningModel
-from my_model.detector.object_detection import ObjectDetector
+from my_model.object_detection import ObjectDetector
 import my_model.config.kbvqa_config as config
 
 
@@ -51,7 +51,6 @@ class KBVQA:
 
     def __init__(self):
 
-       # self.col1, self.col2, self.col3 = st.columns([0.2, 0.6, 0.2])
         if st.session_state["method"] == "7b-Fine-Tuned Model":
             self.kbvqa_model_name: str = config.KBVQA_MODEL_NAME_7b
         elif st.session_state["method"] == "13b-Fine-Tuned Model":