Update app.py

app.py

@@ -53,3 +53,15 @@ image = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
 
 # Text input for the question
 question = st.text_input("Enter your question about the image:")
+
+
+if st.button("Get Answer"):
+    if image is not None and question:
+        # Display the image
+        st.image(image, use_column_width=True)
+        # Get and display the answer
+        model, processor = load_caption_model()
+        answer = answer_question(image, question, model, processor)
+        st.write(answer)
+    else:
+        st.write("Please upload an image and enter a question.")