| import pickle | |
| import faiss | |
| import streamlit as st | |
| from sentence_transformers import SentenceTransformer | |
| def show_search(model, faiss_index, sentences): | |
| query = st.text_input("Enter your search query:") | |
| if query: | |
| # Convert query to embedding | |
| query_embedding = model.encode([query])[0].reshape(1, -1) | |
| # Perform search | |
| D, I = faiss_index.search(query_embedding, k=5) # Search for top 5 similar items | |
| # Display results | |
| st.write("Top results:") | |
| for i in I[0]: | |
| st.write(sentences[i]) |