Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -159,12 +159,12 @@ class KBVQA():
 
         return output_text.capitalize()
 
-def prepare_kbvqa_model(detection_model, only_reload_detection_model=False):
+def prepare_kbvqa_model(only_reload_detection_model=False):
     free_gpu_resources()
     kbvqa = KBVQA()
-    kbvqa.detection_model = detection_model
+    kbvqa.detection_model = st.session_state.detection_model
     # Progress bar for model loading
-    with st.spinner('Loading model...'):
+    with st.spinner('Loading model... this should take no more than a few minutes.'):
 
         if not only_reload_detection_model:
             progress_bar = st.progress(0)