Update app.py

app.py

@@ -22,7 +22,7 @@ class ImageHandler:
         caption = model.get_caption(img)
         image_with_boxes, detected_objects_str = model.detect_objects(img)
         if show_processed_image:
-            st.image(image_with_boxes)
+            st.image(image_with_boxes, use_column_width=True)
         return caption, detected_objects_str
 
     @staticmethod
@@ -36,7 +36,7 @@ class QuestionAnswering:
     @staticmethod
     def answer_question(image, question, caption, detected_objects_str, model):
         answer = model.generate_answer(question, caption, detected_objects_str)
-        st.image(image)
+        st.image(image, use_column_width=True)
         st.write(caption)
         st.write("----------------")
         st.write(detected_objects_str)
@@ -58,7 +58,7 @@ class UIComponents:
 
     @staticmethod
     def display_image(img):
-        st.image(img)
+        st.image(img, use_column_width=True)
 
 def load_kbvqa_model(detection_model):
     """Load KBVQA Model based on the selected detection model."""
@@ -100,6 +100,9 @@ def image_qa_app(kbvqa_model):
     if 'current_image' in st.session_state:
         UIComponents.display_image(st.session_state['current_image'])
 
+    else:
+        st.write("image not uploaded properly")
+