m7mdal7aj commited on
Commit
45969bf
1 Parent(s): 8cf7678

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -75,6 +75,7 @@ def image_qa_app(kbvqa):
75
  if st.button('Get Answer'):
76
  # Process the question
77
  answer = answer_question(st.session_state['current_image'], question, model=kbvqa)
 
78
  st.session_state['qa_history'].append((question, answer))
79
 
80
  # Display all Q&A
 
75
  if st.button('Get Answer'):
76
  # Process the question
77
  answer = answer_question(st.session_state['current_image'], question, model=kbvqa)
78
+ free_gpu_resources()
79
  st.session_state['qa_history'].append((question, answer))
80
 
81
  # Display all Q&A