Update my_model/tabs/run_inference.py
Browse files
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 =
|
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
|