Update my_model/KBVQA.py
Browse files- my_model/KBVQA.py +2 -1
my_model/KBVQA.py
CHANGED
@@ -267,9 +267,10 @@ def prepare_kbvqa_model(only_reload_detection_model: bool = False) -> KBVQA:
|
|
267 |
kbvqa = KBVQA()
|
268 |
kbvqa.detection_model = st.session_state.detection_model
|
269 |
# Progress bar for model loading
|
270 |
-
with st.spinner('Loading model...
|
271 |
|
272 |
if not only_reload_detection_model:
|
|
|
273 |
progress_bar = st.progress(0)
|
274 |
kbvqa.load_detector(kbvqa.detection_model)
|
275 |
progress_bar.progress(33)
|
|
|
267 |
kbvqa = KBVQA()
|
268 |
kbvqa.detection_model = st.session_state.detection_model
|
269 |
# Progress bar for model loading
|
270 |
+
with st.spinner('Loading model...'):
|
271 |
|
272 |
if not only_reload_detection_model:
|
273 |
+
st.text('this should take no more than a few minutes!')
|
274 |
progress_bar = st.progress(0)
|
275 |
kbvqa.load_detector(kbvqa.detection_model)
|
276 |
progress_bar.progress(33)
|