Update app.py

app.py

@@ -52,7 +52,6 @@ def run_inference():
         image_qa_app(st.session_state['kbvqa'])
 
 
-
 def image_qa_app(kbvqa):
     # Initialize session state for storing the current image and its Q&A history
     if 'current_image' not in st.session_state:
@@ -66,7 +65,8 @@ def image_qa_app(kbvqa):
     for idx, sample_image_path in enumerate(sample_images):
         with cols[idx]:
             image = Image.open(sample_image_path)
-            if st.image(image, use_column_width=True):
+            # Use button for each image
+            if st.button(f'Sample Image {idx + 1}', key=f'sample_{idx}'):
                 st.session_state['current_image'] = image
                 st.session_state['qa_history'] = []
 
@@ -93,7 +93,7 @@ def image_qa_app(kbvqa):
         # Display all Q&A
         for q, a in st.session_state['qa_history']:
             st.text(f"Q: {q}\nA: {a}\n")
-            
+
             
 # Main function
 def main():