Update README.md

example code to do inference

README.md

@@ -27,9 +27,31 @@ I created at `phunc20/nj_biergarten_captcha_v2`.
 
 ### Direct Use
 
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+```python
+import torch
 
-[More Information Needed]
+if torch.cuda.is_available():
+    device = torch.device("cuda")
+else:
+    device = torch.device("cpu")
+
+
+from transformers import TrOCRProcessor, VisionEncoderDecoderModel
+
+hub_dir = "phunc20/trocr-base-handwritten_nj_biergarten_captcha_v2"
+processor = TrOCRProcessor.from_pretrained(hub_dir)
+model = VisionEncoderDecoderModel.from_pretrained(hub_dir)
+model = model.to(device)
+
+
+from PIL import Image
+
+image = Image.open("/path/to/image")
+pixel_values = processor(image, return_tensors='pt').pixel_values
+pixel_values = pixel_values.to(device)
+outputs = model.generate(pixel_values)
+pred_str = processor.batch_decode(outputs, skip_special_tokens=True)[0]
+```
 
 
 ## Bias, Risks, and Limitations