m7mdal7aj commited on
Commit
c62c890
1 Parent(s): 2b41e89

Update my_model/KBVQA.py

Browse files
Files changed (1) hide show
  1. 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