m7mdal7aj commited on
Commit
df43e4c
1 Parent(s): d182243

Update my_model/KBVQA.py

Browse files
Files changed (1) hide show
  1. my_model/KBVQA.py +1 -3
my_model/KBVQA.py CHANGED
@@ -168,13 +168,11 @@ def prepare_kbvqa_model(only_reload_detection_model=False):
168
 
169
  if not only_reload_detection_model:
170
  progress_bar = st.progress(0)
171
- st.text("BBBBBBB")
172
  kbvqa.load_detector(kbvqa.detection_model)
173
- st.text("CCCCCC")
174
  progress_bar.progress(33)
175
  kbvqa.load_caption_model()
176
  free_gpu_resources()
177
- progress_bar.progress(66)
178
  kbvqa.load_fine_tuned_model()
179
  free_gpu_resources()
180
  progress_bar.progress(100)
 
168
 
169
  if not only_reload_detection_model:
170
  progress_bar = st.progress(0)
 
171
  kbvqa.load_detector(kbvqa.detection_model)
 
172
  progress_bar.progress(33)
173
  kbvqa.load_caption_model()
174
  free_gpu_resources()
175
+ progress_bar.progress(75)
176
  kbvqa.load_fine_tuned_model()
177
  free_gpu_resources()
178
  progress_bar.progress(100)