zenoda commited on
Commit
afd5745
1 Parent(s): 649e257

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
README.md ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ```
2
+ from transformers import VisionEncoderDecoderModel, TrOCRProcessor
3
+ from PIL import Image
4
+ import requests
5
+
6
+ processor = TrOCRProcessor.from_pretrained("zenoda/trocr-captcha-killer")
7
+ model = VisionEncoderDecoderModel.from_pretrained("zenoda/trocr-captcha-killer")
8
+ model.to('cuda')
9
+ url = 'https://huggingface.co/datasets/zenoda/trocr-captcha-killer/resolve/main/106-1688354008849.png'
10
+ image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
11
+ generated_ids = model.generate(processor(image, return_tensors="pt").pixel_values.to('cuda'))
12
+ predictText = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
13
+ print(predictText)
14
+
15
+ ```