Update src/streamlit_app.py

src/streamlit_app.py

@@ -166,13 +166,14 @@ The goal of this system is to retrieve the most relevant fashion images based on
 # Example to show some images
 st.subheader("Some Sample Images and Their Captions:")
 example = [13422, 10037, 38246, 23273, 2008]
+cols = st.columns(5)  # Chia thành 5 cột
 for idx, img_id in enumerate(example):
-    show_img(img_id, None, idx)
+    show_img(img_id, None, cols[idx])
 
 # Chạy ví dụ với prompt
 st.subheader("Example Usage: Enter a Prompt to Retrieve Related Images")
 prompt_input = st.text_input("Enter a prompt", "a red Apparel dress")
 top_k_input = st.number_input("Enter the number of results (top_k)", min_value=1, max_value=10, value=5)
 
-if prompt_input:
+if st.button("Find top 5 related images:"):
     running(prompt_input, top_k_input)