Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -164,7 +164,7 @@ def prepare_kbvqa_model(only_reload_detection_model=False):
     kbvqa = KBVQA()
     kbvqa.detection_model = st.session_state.detection_model
     # Progress bar for model loading
-    with st.spinner('Loading model... this should take no more than a few minutes.'):
+    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)
@@ -173,6 +173,7 @@ def prepare_kbvqa_model(only_reload_detection_model=False):
             kbvqa.load_caption_model()
             free_gpu_resources()
             progress_bar.progress(75)
+            st.text('Almost there :)')
             kbvqa.load_fine_tuned_model()
             free_gpu_resources()
             progress_bar.progress(100)