Update src/streamlit_app.py

src/streamlit_app.py

@@ -61,6 +61,8 @@ def show_img(img_id, score=None, col=None):
                 # Hiển thị ảnh trong cột
                 if col:
                     col.image(img, caption=text, use_container_width=True)
+            else:
+                st.write("img_style is empty")
 
         except requests.exceptions.RequestException as e:
             st.error(f"Error fetching image: {e}")
@@ -84,7 +86,7 @@ def search_faiss(model, processor, index, id_map, prompt, top_k=5, device='cpu')
     st.write("FAISS search completed.")
     st.write("Indices returned:", I[0])
     st.write("Scores returned:", D[0])
-    st.write("ID map keys sample:", list(id_map.keys())[:10])
+    # st.write("ID map keys sample:", list(id_map.keys())[:10])
     return [(id_map[i], float(D[0][j])) for j, i in enumerate(I[0])]
 
 def running(prompt, top_k=5):