Spaces:
Sleeping
Sleeping
Download Src/retrieval/test.py from kanhacoderx/Documind-AI: direct link, hf CLI and curl.
- Browser
- Download file 398 Bytes
-
https://huggingface.co/spaces/kanhacoderx/Documind-AI/resolve/main/Src/retrieval/test.py
- Command line
-
hf download hf://spaces/kanhacoderx/Documind-AI/Src/retrieval/test.py
-
curl -L -o test.py https://huggingface.co/spaces/kanhacoderx/Documind-AI/resolve/main/Src/retrieval/test.py
398 Bytes
| from Src.embeddings.embedder import Embedder | |
| from Src.vectorstore.faiss_store import FAISSStore | |
| from Src.retrieval.retriever import Retriever | |
| #Load Vector store | |
| embedder=Embedder() | |
| faiss_store=FAISSStore(embedder.embedding_model) | |
| faiss_store.load_vector_store() | |
| retriever=Retriever(faiss_store.vector_store) | |
| query='What Is The Main Idea Of Document' | |
| context=retriever.retrieve(query) |