guptarohit20/NVIDIA-NIM-Demo
0
NVIDIA DeepSeek AI Document Search
Overview ๐ง ๐
This project is an AI-powered document retrieval and question-answering system utilizing NVIDIA DeepSeek AI and FAISS vector stores. It allows users to embed, retrieve, and query research papers using advanced NVIDIA AI models for accurate and contextual responses.
Features โจ
- FAISS-based Vector Storage: Efficiently stores and retrieves document embeddings.
- NVIDIA DeepSeek AI Integration: Uses deepseek-ai/deepseek-r1 for high-quality AI inference.
- PDF Processing: Extracts and processes research papers for retrieval-based QA.
- Streamlit UI: Interactive user interface for querying and document similarity search.
Deployment ๐
The application is deployed on Hugging Face Spaces. You can access it using the following link:
๐ [Deployment Link](https://huggingface.co/spaces/guptarohit20/NVIDIA-NIM-Demo)
Installation & Setup ๐ ๏ธ
1๏ธโฃ Clone the Repository
git clone https://github.com/Rohit-Madhesiya/NVIDIA-NIM-Demo.git
cd NVIDIA-NIM-Demo2๏ธโฃ Create a Virtual Environment (Optional but Recommended)
python -m venv venv
source venv/bin/activate # Windows: venv\Scripts\activate3๏ธโฃ Install Dependencies
pip install -r requirements.txt4๏ธโฃ Set Up API Key
- Create a
.envfile in the root directory. - Add the following line:
NVIDIA_API_KEY=<your_nvidia_api_key>Usage ๐
Run the Streamlit application:
streamlit run main.pyOnce the app is running:
- Enter your NVIDIA API Key.
- Click Document Embedding to process research papers.
- Type a question and get AI-generated responses based on the document content.
- Expand Document Similarity Search to view retrieved document chunks.
Dependencies ๐
The project requires the following Python libraries:
streamlitlangchain_nvidia_ai_endpointslangchain_communityfaiss-cpupypdfpython-dotenvopenai
Project Structure ๐
โโโ main.py # Main application file
โโโ requirements.txt # List of dependencies
โโโ .env # API key configuration (not included in repo)
โโโ README.md # Project documentationContributing ๐ค
Feel free to fork the repository, submit pull requests, or report any issues.
Author ๐จโ๐ป
Developed by [Rohit Gupta].
