m7mdal7aj commited on
Commit
95cc577
·
verified ·
1 Parent(s): 814ed9a

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +1 -0
my_model/tabs/run_inference.py CHANGED
@@ -301,6 +301,7 @@ class InferenceRunner(StateManager):
301
 
302
  st.write(st.session_state['method'])
303
  st.write(st.session_state['previous_state']['method'])
 
304
 
305
  if self.is_model_loaded:
306
  free_gpu_resources()
 
301
 
302
  st.write(st.session_state['method'])
303
  st.write(st.session_state['previous_state']['method'])
304
+ st.write(st.session_state['kbvqa'].kbvqa_model_name
305
 
306
  if self.is_model_loaded:
307
  free_gpu_resources()