jkorstad commited on
Commit
b2dd8bf
·
verified ·
1 Parent(s): 83f10d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,12 +3,14 @@ from transformers import AutoProcessor, AutoModelForCausalLM
3
  from pdf2image import convert_from_path
4
  import base64
5
  import io
 
6
  from PIL import Image
7
 
8
  # Load the OCR model and processor from Hugging Face
9
  processor = AutoProcessor.from_pretrained("allenai/olmOCR-7B-0225-preview")
10
  model = AutoModelForCausalLM.from_pretrained("allenai/olmOCR-7B-0225-preview")
11
 
 
12
  def process_pdf(pdf_file):
13
  """
14
  Process the uploaded PDF file, extract text from each page, and generate HTML
 
3
  from pdf2image import convert_from_path
4
  import base64
5
  import io
6
+ import spaces
7
  from PIL import Image
8
 
9
  # Load the OCR model and processor from Hugging Face
10
  processor = AutoProcessor.from_pretrained("allenai/olmOCR-7B-0225-preview")
11
  model = AutoModelForCausalLM.from_pretrained("allenai/olmOCR-7B-0225-preview")
12
 
13
+ @spaces.GPU
14
  def process_pdf(pdf_file):
15
  """
16
  Process the uploaded PDF file, extract text from each page, and generate HTML