Spaces:

tensorlake
/

document-extractors

Sleeping

rishiraj commited on Jun 7

Commit

b3302d2

•

1 Parent(s): 33f6a35

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,6 +131,7 @@ with gr.Blocks(title="PDF data extraction with PDF Extractor & Indexify") as pdf
 		outputs = [model_output_text_box]
 	)
 def use_gemini(pdf_filepath, key):
 	if pdf_filepath is None:
 		raise gr.Error("Please provide some input PDF: upload an PDF file")
@@ -197,6 +198,7 @@ with gr.Blocks(title="PDF data extraction with Gemini & Indexify") as gemini_dem
 		outputs = [model_output_text_box]
 	)
 def use_openai(pdf_filepath, key):
 	if pdf_filepath is None:
 		raise gr.Error("Please provide some input PDF: upload an PDF file")

 		outputs = [model_output_text_box]
 	)
+@spaces.GPU
 def use_gemini(pdf_filepath, key):
 	if pdf_filepath is None:
 		raise gr.Error("Please provide some input PDF: upload an PDF file")
 		outputs = [model_output_text_box]
 	)
+@spaces.GPU
 def use_openai(pdf_filepath, key):
 	if pdf_filepath is None:
 		raise gr.Error("Please provide some input PDF: upload an PDF file")