Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ hybrid_rag = HybridColpaliRAG(
|
|
67 |
IngestResult = namedtuple("IngestResult", ["status_text", "progress_table"])
|
68 |
|
69 |
|
70 |
-
@spaces.GPU(duration=
|
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=
|
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 = {}
|