Update Home.py

Home.py

@@ -8,14 +8,16 @@ from transformers import pipeline, Conversation
 # txtgen = pipeline(task="text-generation", model="EleutherAI/gpt-neo-2.7B")
 # txtclassifi = pipeline(task="text-classification", model="nlptown/bert-base-multilingual-uncased-sentiment")
 # summurize = pipeline(task="summarization", model="sshleifer/distilbart-cnn-12-6")
-visualqna = pipeline(task="vqa", model="microsoft/DialoGPT-medium")
+# visualqna = pipeline(task="vqa", model="microsoft/DialoGPT-medium")
+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.image(img)
+            st.write("Image Uploaded!")
+            # st.image(img)
     return img
 
 # def homepage():
@@ -79,6 +81,7 @@ 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)