Spaces:

nhay103
/

VietOCR

Runtime error

nhay103 commited on Aug 6, 2023

Commit

e5923d4

•

1 Parent(s): 57a6602

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ model = VietOCR(len(vocab),
         config['cnn'],
         config['transformer'],
         config['seq_modeling'])
-model.load_state_dict(torch.load('train_old.pth'))
 def predict(inp):
     img = process_input(inp, config['dataset']['image_height'],

         config['cnn'],
         config['transformer'],
         config['seq_modeling'])
+model.load_state_dict(torch.load('train_old.pth', map_location=torch.device('cpu')))
 def predict(inp):
     img = process_input(inp, config['dataset']['image_height'],