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