Spaces:
Running
on
Zero
Running
on
Zero
jedick
commited on
Commit
·
59eeb53
1
Parent(s):
5cdd81a
Add @spaces.GPU() decorator to retrieval function
Browse files
app.py
CHANGED
@@ -336,6 +336,7 @@ with gr.Blocks(theme=my_theme, css=custom_css, head=font_awesome_html) as demo:
|
|
336 |
pdf_file = f"examples/retrieval/{pdf_file}"
|
337 |
return pdf_file, claim
|
338 |
|
|
|
339 |
def retrieve_evidence_with_method(pdf_file, claim, top_k, method):
|
340 |
"""
|
341 |
Retrieve evidence using the selected method
|
|
|
336 |
pdf_file = f"examples/retrieval/{pdf_file}"
|
337 |
return pdf_file, claim
|
338 |
|
339 |
+
@spaces.GPU()
|
340 |
def retrieve_evidence_with_method(pdf_file, claim, top_k, method):
|
341 |
"""
|
342 |
Retrieve evidence using the selected method
|