Update my_model/KBVQA.py
Browse files- my_model/KBVQA.py +1 -1
my_model/KBVQA.py
CHANGED
@@ -145,7 +145,7 @@ def prepare_kbvqa_model(detection_model):
|
|
145 |
free_gpu_resources()
|
146 |
kbvqa = KBVQA()
|
147 |
# Progress bar for model loading
|
148 |
-
with st.spinner('Loading
|
149 |
|
150 |
progress_bar = st.progress(0)
|
151 |
kbvqa.load_detector(detection_model)
|
|
|
145 |
free_gpu_resources()
|
146 |
kbvqa = KBVQA()
|
147 |
# Progress bar for model loading
|
148 |
+
with st.spinner('Loading model...'):
|
149 |
|
150 |
progress_bar = st.progress(0)
|
151 |
kbvqa.load_detector(detection_model)
|