Spaces:
Sleeping
Sleeping
Update my_model/KBVQA.py
Browse files- my_model/KBVQA.py +1 -1
my_model/KBVQA.py
CHANGED
@@ -184,7 +184,7 @@ def prepare_kbvqa_model(detection_model, only_reload_detection_model=False):
|
|
184 |
progress_bar.progress(100)
|
185 |
|
186 |
if kbvqa.all_models_loaded:
|
187 |
-
st.success('Model loaded successfully!')
|
188 |
kbvqa.kbvqa_model.eval()
|
189 |
free_gpu_resources()
|
190 |
return kbvqa
|
|
|
184 |
progress_bar.progress(100)
|
185 |
|
186 |
if kbvqa.all_models_loaded:
|
187 |
+
st.success('Model loaded successfully and ready for inferecne!')
|
188 |
kbvqa.kbvqa_model.eval()
|
189 |
free_gpu_resources()
|
190 |
return kbvqa
|