Update my_model/tabs/run_inference.py

my_model/tabs/run_inference.py

@@ -9,8 +9,7 @@ import torch.nn as nn
 import pandas as pd
 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.gen_utilities import free_gpu_resources
 from my_model.state_manager import StateManager
 
 state_manager = StateManager()
@@ -71,7 +70,6 @@ class InferenceRunner(StateManager):
                 st.text(f"Q: {q}\nA: {a}\n")
 
 
-
     def run_inference(self):
         st.title("Run Inference")
         self.initialize_state()