AdithyaSK commited on
Commit
ad6ac6f
1 Parent(s): 30cca2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -67,7 +67,7 @@ hybrid_rag = HybridColpaliRAG(
67
  IngestResult = namedtuple("IngestResult", ["status_text", "progress_table"])
68
 
69
 
70
- @spaces.GPU(duration=600)
71
  def ingest_data(pdf_files, use_ocr, chunk_size, progress=gr.Progress()):
72
  file_paths = [pdf_file.name for pdf_file in pdf_files]
73
  total_start_time = time.time()
@@ -157,7 +157,7 @@ def ingest_data(pdf_files, use_ocr, chunk_size, progress=gr.Progress()):
157
  )
158
 
159
 
160
- @spaces.GPU(duration=600)
161
  def retrieve_data(query, top_k, sequential=False):
162
  results = {}
163
  timings = {}
 
67
  IngestResult = namedtuple("IngestResult", ["status_text", "progress_table"])
68
 
69
 
70
+ @spaces.GPU(duration=120)
71
  def ingest_data(pdf_files, use_ocr, chunk_size, progress=gr.Progress()):
72
  file_paths = [pdf_file.name for pdf_file in pdf_files]
73
  total_start_time = time.time()
 
157
  )
158
 
159
 
160
+ @spaces.GPU(duration=120)
161
  def retrieve_data(query, top_k, sequential=False):
162
  results = {}
163
  timings = {}