Colpali RAG

Colpali RAG using VARAG¶

Requirement to RUN this notebook - Min T4 GPU

In [ ]:

!git clone https://github.com/adithya-s-k/VARAG
%cd VARAG
%pwd

!git clone https://github.com/adithya-s-k/VARAG %cd VARAG %pwd

In [ ]:

!apt-get update && apt-get install -y && apt-get install -y poppler-utils

!apt-get update && apt-get install -y && apt-get install -y poppler-utils

In [ ]:

%pip install -e .

%pip install -e .

In [ ]:

from sentence_transformers import SentenceTransformer
from varag.rag import ColpaliRAG
from varag.llms import OpenAI
from varag.utils import get_model_colpali
import lancedb
import os
from dotenv import load_dotenv

os.environ["OPENAI_API_KEY"] = "api-key"

load_dotenv()

from sentence_transformers import SentenceTransformer from varag.rag import ColpaliRAG from varag.llms import OpenAI from varag.utils import get_model_colpali import lancedb import os from dotenv import load_dotenv os.environ["OPENAI_API_KEY"] = "api-key" load_dotenv()

In [ ]:

shared_db = lancedb.connect("~/shared_rag_db")

model, processor = get_model_colpali("vidore/colpali-v1.2")

colpali_rag = ColpaliRAG(
    colpali_model=model,
    colpali_processor=processor,
    db=shared_db,
    table_name="colpaliDemo",
)

vlm = OpenAI()

shared_db = lancedb.connect("~/shared_rag_db") model, processor = get_model_colpali("vidore/colpali-v1.2") colpali_rag = ColpaliRAG( colpali_model=model, colpali_processor=processor, db=shared_db, table_name="colpaliDemo", ) vlm = OpenAI()

In [ ]:

colpali_rag.index(
        "./examples/data", 
        overwrite=False, 
        recursive=False, 
        verbose=True
    )

colpali_rag.index( "./examples/data", overwrite=False, recursive=False, verbose=True )

In [ ]:

query = "What is Colpali"
num_results = 5

results = colpali_rag.search(query, k=5)

images = [result["image"] for result in results]

# Display the images
for i, img in enumerate(images, 1):
    print(f"Image {i}:")
    display(img)

query = "What is Colpali" num_results = 5 results = colpali_rag.search(query, k=5) images = [result["image"] for result in results] # Display the images for i, img in enumerate(images, 1): print(f"Image {i}:") display(img)

In [ ]:

from IPython.display import display, Markdown


response = vlm.query(query, images, max_tokens=1000)


display(Markdown(response))

from IPython.display import display, Markdown response = vlm.query(query, images, max_tokens=1000) display(Markdown(response))

Run Gradio Demo¶

In [ ]:

%cd examples
!python colpaliDemo.py --share

%cd examples !python colpaliDemo.py --share