Update app.py

app.py

@@ -89,13 +89,15 @@ def image_qa_app(kbvqa_model):
     UIComponents.display_image_selection(sample_images)
 
     uploaded_image = st.file_uploader("Or upload an Image", type=["png", "jpg", "jpeg"])
-    UIComponents.display_image(uploaded_image)
+    
     if uploaded_image is not None:
         st.session_state['current_image'] = Image.open(uploaded_image)
         st.session_state['qa_history'] = []
         st.session_state['analysis_done'] = False
         st.session_state['answer_in_progress'] = False
 
+        UIComponents.display_image(st.session_state['current_image'])
+
     if st.session_state.get('current_image') and not st.session_state.get('analysis_done', False):
         if st.button('Analyze Image'):
             caption, detected_objects_str = ImageHandler.analyze_image(st.session_state['current_image'], kbvqa_model)