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
|