jedick commited on
Commit
59eeb53
·
1 Parent(s): 5cdd81a

Add @spaces.GPU() decorator to retrieval function

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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