Update my_model/KBVQA.py
Browse files- my_model/KBVQA.py +2 -1
my_model/KBVQA.py
CHANGED
@@ -168,8 +168,9 @@ def prepare_kbvqa_model(only_reload_detection_model=False):
|
|
168 |
|
169 |
if not only_reload_detection_model:
|
170 |
progress_bar = st.progress(0)
|
171 |
-
|
172 |
kbvqa.load_detector(kbvqa.detection_model)
|
|
|
173 |
progress_bar.progress(33)
|
174 |
kbvqa.load_caption_model()
|
175 |
free_gpu_resources()
|
|
|
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()
|