m7mdal7aj commited on
Commit
8024886
1 Parent(s): 0b430a0

Update my_model/KBVQA.py

Browse files
Files changed (1) hide show
  1. 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... this should take no more than a few minutes!'):
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)