Hugging Face's logo

microsoft
/
dit-base

Transformers
PyTorch
beit
dit
Model card Files
xet
Community
4
nielsr HF Staff commited on
Commit
1ca7ca1
·
1 Parent(s): f521936

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ from transformers import BeitFeatureExtractor, BeitForMaskedImageModeling
29
  import torch
30
  from PIL import Image
31
 
32
- image = Image.open('path_to_your_document_image')
33
 
34
  feature_extractor = BeitFeatureExtractor.from_pretrained("microsoft/dit-base")
35
  model = BeitForMaskedImageModeling.from_pretrained("microsoft/dit-base")
 
29
  import torch
30
  from PIL import Image
31
 
32
+ image = Image.open('path_to_your_document_image').convert('RGB')
33
 
34
  feature_extractor = BeitFeatureExtractor.from_pretrained("microsoft/dit-base")
35
  model = BeitForMaskedImageModeling.from_pretrained("microsoft/dit-base")