Genzo1010 commited on
Commit
25e1f2b
·
verified ·
1 Parent(s): ef0b922

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
  # Example for PyTorch model or doctr model
18
 
19
- ocr_model = ocr_predictor(det_arch='crnn_mobilenet_v3_small', reco_arch='crnn_mobilenet_v3_small', pretrained=True).to(device)
20
 
21
 
22
  import torch
 
16
 
17
  # Example for PyTorch model or doctr model
18
 
19
+ ocr_model = ocr_predictor(det_arch='db_mobilenet_v3_large', reco_arch='crnn_mobilenet_v3_small', pretrained=True).to(device)
20
 
21
 
22
  import torch