Update app.py

app.py

@@ -89,15 +89,21 @@ 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"])
-    
+
     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
 
+    # Display the image if it's in the session state
+    if 'current_image' in st.session_state:
         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)