Spaces:
Running
Running
File size: 464 Bytes
9ff79dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import requests
from PIL import Image
def load_from_pdf(pdf_path: str):
from pdf2image import convert_from_path
images = convert_from_path(pdf_path)
return images
def load_from_image_urls(urls: str):
images = [Image.open(requests.get(url, stream=True).raw) for url in urls]
return images
def load_from_dataset(dataset):
from datasets import load_dataset
dataset = load_dataset(dataset, split="test")
return dataset["image"]
|