Update app.py

app.py

@@ -30,6 +30,8 @@ class ImageHandler:
         # Implementation for freeing GPU resources
         free_gpu_resources()
 
+
+
 class QuestionAnswering:
     @staticmethod
     def answer_question(image, question, caption, detected_objects_str, model):
@@ -54,6 +56,10 @@ class UIComponents:
                     st.session_state['analysis_done'] = False
                     st.session_state['answer_in_progress'] = False
 
+    @staticmethod
+    def display_image(img):
+        st.image(img)
+
 def load_kbvqa_model(detection_model):
     """Load KBVQA Model based on the selected detection model."""
     if st.session_state.get('kbvqa') is not None:
@@ -83,6 +89,7 @@ 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'] = []
@@ -131,11 +138,11 @@ def run_inference():
         if 'kbvqa' not in st.session_state or st.session_state['detection_model'] != detection_model:
             st.session_state['detection_model'] = detection_model
             if 'model' not in st.session_state:
-            if st.button('Load Model'):
+                if st.button('Load Model'):
 
-                if load_kbvqa_model(detection_model):
-                    set_model_confidence(detection_model)
-                    image_qa_app(st.session_state['kbvqa'])
+                    if load_kbvqa_model(detection_model):
+                        set_model_confidence(detection_model)
+                        image_qa_app(st.session_state['kbvqa'])