KB-VQA

Sleeping

m7mdal7aj commited on May 15

Commit

a70e150

•

1 Parent(s): 95cc577

Update my_model/tabs/run_inference.py

Files changed (1) hide show

my_model/tabs/run_inference.py CHANGED Viewed

@@ -301,7 +301,7 @@ class InferenceRunner(StateManager):
             st.write(st.session_state['method'])
             st.write(st.session_state['previous_state']['method'])
-            st.write(st.session_state['kbvqa'].kbvqa_model_name
         if self.is_model_loaded:
             free_gpu_resources()

             st.write(st.session_state['method'])
             st.write(st.session_state['previous_state']['method'])
+            st.write(st.session_state['kbvqa'].kbvqa_model_name)
         if self.is_model_loaded:
             free_gpu_resources()