Spaces:
Runtime error
Runtime error
File size: 512 Bytes
eaa18ed adfd1fe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from PIL import Image
import requests
from io import BytesIO
image_url = "https://raw.githubusercontent.com/scb10x/typhoon-ocr/main/assets/test-image.png"
response = requests.get(image_url)
image = Image.open(BytesIO(response.content)).convert("RGB")
inputs = processor(images=image, return_tensors="pt")
inputs = {k: v.to(device) for k, v in inputs.items()}
generated_ids = model.generate(**inputs, max_new_tokens=256)
result = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
print(result)
|