Spaces:
Sleeping
Sleeping
File size: 809 Bytes
812c941 ee142e9 812c941 ee142e9 812c941 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import streamlit
from PIL import Image
import torch
from transformers import TrOCRProcessor, VisionEncoderDecoderModel
MODEL_PATH = 'model/'
processor = TrOCRProcessor.from_pretrained(MODEL_PATH)
model = VisionEncoderDecoderModel.from_pretrained(MODEL_PATH)
streamlit.title('Light OCR')
uploaded_file = streamlit.file_uploader('Choose an image...', type=['png', 'jpg', 'jpeg'])
if uploaded_file:
image = Image.open(uploaded_file).convert('RGB')
streamlit.image(image, caption='Uploaded Image', use_container_width=True)
pixel_values = processor(images=image, return_tensors='pt').pixel_values
output_ids = model.generate(pixel_values)
text = processor.batch_decode(output_ids, skip_special_tokens=True)[0]
streamlit.subheader('Recognized Text')
streamlit.write(text)
|