m7mdal7aj commited on
Commit
6657600
1 Parent(s): 1a4044e

Update my_model/tabs/run_inference.py

Browse files
Files changed (1) hide show
  1. my_model/tabs/run_inference.py +1 -1
my_model/tabs/run_inference.py CHANGED
@@ -47,7 +47,7 @@ class InferenceRunner(StateManager):
47
  str: The generated answer to the question.
48
  """
49
  free_gpu_resources()
50
- answer = self.session_state.kbvqa.generate_answer(question, caption, detected_objects_str)
51
  prompt_length = model.current_prompt_length
52
  free_gpu_resources()
53
  return answer, prompt_length
 
47
  str: The generated answer to the question.
48
  """
49
  free_gpu_resources()
50
+ answer = st.session_state.kbvqa.generate_answer(question, caption, detected_objects_str)
51
  prompt_length = model.current_prompt_length
52
  free_gpu_resources()
53
  return answer, prompt_length