Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
|
@@ -48,7 +48,7 @@ class InferenceRunner(StateManager):
|
|
| 48 |
"""
|
| 49 |
free_gpu_resources()
|
| 50 |
answer = st.session_state.kbvqa.generate_answer(question, caption, detected_objects_str)
|
| 51 |
-
prompt_length =
|
| 52 |
free_gpu_resources()
|
| 53 |
return answer, prompt_length
|
| 54 |
|
|
|
|
| 48 |
"""
|
| 49 |
free_gpu_resources()
|
| 50 |
answer = st.session_state.kbvqa.generate_answer(question, caption, detected_objects_str)
|
| 51 |
+
prompt_length = st.session_state.kbvqa.current_prompt_length
|
| 52 |
free_gpu_resources()
|
| 53 |
return answer, prompt_length
|
| 54 |
|