Update app.py

app.py

@@ -67,7 +67,7 @@ def convert_files(files):
 
 @spaces.GPU
 def index_gpu(images, ds):
-    """Example script to run inference with ColPali"""
+    """Example script to run inference with ColPali (ColQwen2)"""
 
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
     if device != model.device:
@@ -91,9 +91,8 @@ def index_gpu(images, ds):
 
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# ColPali: Efficient Document Retrieval with Vision Language Models 📚")
-    gr.Markdown("""Demo to test ColPali on PDF documents. The inference code is based on the [ViDoRe benchmark](https://github.com/illuin-tech/vidore-benchmark).
-
+    gr.Markdown("# ColPali: Efficient Document Retrieval with Vision Language Models (ColQwen2) 📚")
+    gr.Markdown("""Demo to test ColQwen2 (ColPali) on PDF documents. 
     ColPali is model implemented from the [ColPali paper](https://arxiv.org/abs/2407.01449).
 
     This demo allows you to upload PDF files and search for the most relevant pages based on your query.