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 = {}
|