Update app.py

app.py

@@ -18,6 +18,7 @@ from varag.utils import get_model_colpali
 import argparse
 import spaces
 import torch
+from docling.document_converter import DocumentConverter
 
 load_dotenv()
 
@@ -39,7 +40,7 @@ image_embedding_model = SentenceTransformer(
 )
 colpali_model, colpali_processor = get_model_colpali("vidore/colpali-v1.2")
 
-
+converter = DocumentConverter()
 
 # Initialize RAG instances
 simple_rag = SimpleRAG(
@@ -66,7 +67,7 @@ hybrid_rag = HybridColpaliRAG(
 IngestResult = namedtuple("IngestResult", ["status_text", "progress_table"])
 
 
-@spaces.GPU
+@spaces.GPU(duration=600)
 def ingest_data(pdf_files, use_ocr, chunk_size, progress=gr.Progress()):
     file_paths = [pdf_file.name for pdf_file in pdf_files]
     total_start_time = time.time()
@@ -156,7 +157,7 @@ def ingest_data(pdf_files, use_ocr, chunk_size, progress=gr.Progress()):
     )
 
 
-@spaces.GPU
+@spaces.GPU(duration=600)
 def retrieve_data(query, top_k, sequential=False):
     results = {}
     timings = {}