Spaces:
Sleeping
Sleeping
Update my_model/KBVQA.py
Browse files- 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(
|
| 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)
|