this

Home.py

@@ -14,10 +14,12 @@ visualqna = pipeline(model="dandelin/vilt-b32-finetuned-vqa")
 
 def load_image():
     with st.sidebar:
-        img = st.text_input("Enter Image URL")
-        if st.button("Load Image"):
-            st.write("Image Uploaded!")
-            # st.image(img)
+        if img := st.text_input("Enter Image URL") or st.selectbox("Select Image", ("Upload Image", "Cancel")):
+            if st.button("Load Image"):
+                st.write("Image Uploaded!")
+                st.image(img)
+        else:
+            st.warning("Please enter an image URL and click 'Load Image' before asking a question.")
     return img
 
 # def homepage():
@@ -81,7 +83,6 @@ def load_image():
 def visual_qna():
     st.title("Visual Q&A")
     img = load_image()
-    st.image(img)
     if img:
         if query := st.chat_input("Enter your message"):
             response = visualqna(question=query, image=img)