arachn0id commited on
Commit
36f94fa
·
1 Parent(s): 5292a0b

working try :)

Browse files
Files changed (2) hide show
  1. app.py +14 -2
  2. requirements.txt +2 -0
app.py CHANGED
@@ -1,4 +1,16 @@
1
  import streamlit as st
2
 
3
- x = st.slider("Select a value")
4
- st.write(x, "squared is", x * x)
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
 
3
+ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
4
+ from PIL import Image
5
+ import requests
6
+
7
+ # load image from the IAM database
8
+ url = "https://fki.tic.heia-fr.ch/static/img/a01-122-02-00.jpg"
9
+ image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
10
+
11
+ processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
12
+ model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
13
+ pixel_values = processor(images=image, return_tensors="pt").pixel_values
14
+
15
+ generated_ids = model.generate(pixel_values)
16
+ generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ torch
2
+ transformers