sergsof2 commited on
Commit
9fc2955
·
verified ·
1 Parent(s): f513898

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ import re
18
  # checkpoint = "/content/drive/MyDrive/T/TrOCR/trocr-small-finetuned"
19
  checkpoint = "microsoft/trocr-small-printed" # если используешь оригинальную base small
20
 
21
- processor = TrOCRProcessor.from_pretrained(checkpoint)
 
22
  model = VisionEncoderDecoderModel.from_pretrained(checkpoint)
23
  model.to('cuda' if torch.cuda.is_available() else 'cpu')
24
  model.eval()
 
18
  # checkpoint = "/content/drive/MyDrive/T/TrOCR/trocr-small-finetuned"
19
  checkpoint = "microsoft/trocr-small-printed" # если используешь оригинальную base small
20
 
21
+ #processor = TrOCRProcessor.from_pretrained(checkpoint)
22
+ processor = TrOCRProcessor.from_pretrained(checkpoint, use_fast=False)
23
  model = VisionEncoderDecoderModel.from_pretrained(checkpoint)
24
  model.to('cuda' if torch.cuda.is_available() else 'cpu')
25
  model.eval()