CoolFace
Apppublic

guptarohit20/NVIDIA-NIM-Demo

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes
App README

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

bash
git clone https://github.com/Rohit-Madhesiya/NVIDIA-NIM-Demo.git
cd NVIDIA-NIM-Demo

2๏ธโƒฃ Create a Virtual Environment (Optional but Recommended)

bash
python -m venv venv
source venv/bin/activate  # Windows: venv\Scripts\activate

3๏ธโƒฃ Install Dependencies

bash
pip install -r requirements.txt

4๏ธโƒฃ Set Up API Key

  • โ€”Create a .env file in the root directory.
  • โ€”Add the following line:
env
NVIDIA_API_KEY=<your_nvidia_api_key>

Usage ๐Ÿ“Œ

Run the Streamlit application:

bash
streamlit run main.py

Once the app is running:

  1. 1.Enter your NVIDIA API Key.
  2. 2.Click Document Embedding to process research papers.
  3. 3.Type a question and get AI-generated responses based on the document content.
  4. 4.Expand Document Similarity Search to view retrieved document chunks.

Dependencies ๐Ÿ“‹

The project requires the following Python libraries:

  • โ€”streamlit
  • โ€”langchain_nvidia_ai_endpoints
  • โ€”langchain_community
  • โ€”faiss-cpu
  • โ€”pypdf
  • โ€”python-dotenv
  • โ€”openai

Project Structure ๐Ÿ“‚

โ”œโ”€โ”€ main.py              # Main application file
โ”œโ”€โ”€ requirements.txt     # List of dependencies
โ”œโ”€โ”€ .env                 # API key configuration (not included in repo)
โ””โ”€โ”€ README.md            # Project documentation

Contributing ๐Ÿค

Feel free to fork the repository, submit pull requests, or report any issues.

Author ๐Ÿ‘จโ€๐Ÿ’ป

Developed by [Rohit Gupta].