Update Home.py

Home.py

@@ -11,6 +11,13 @@ from transformers import pipeline, Conversation
 visualqna = pipeline(task="vqa", model="microsoft/DialoGPT-medium")
 
 
+def load_image():
+    with st.sidebar:
+        img = st.text_input("Enter Image URL")
+        if st.button("Load Image"):
+            st.image(img)
+    return img
+
 # def homepage():
 #     st.write("Timeline")
 #     # allmessages =[]
@@ -71,11 +78,9 @@ visualqna = pipeline(task="vqa", model="microsoft/DialoGPT-medium")
 
 def visual_qna():
     st.title("Visual Q&A")
-    with st.sidebar:
-        if img := st.text_input("Enter Image URL"):
-            st.image(img)
+    img = load_image()
     if query := st.chat_input("Enter your message"):
-        response = visualqna(img, query)
+        response = visualqna(question=query, image=img)
     with st.chat_message("assistant"):
         st.write(response)