Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|