Spaces:

7eu7d7
/

CAPTCHA_recognize

Sleeping

7eu7d7 commited on Jun 5

Commit

5962bdc

1 Parent(s): 1f584ff

bug fix

Files changed (1) hide show

cap.py CHANGED Viewed

@@ -18,7 +18,7 @@ class Predictor:
         self.model = ResnetEncoderDecoder(char_dict).to(device)
         self.model.eval()
         if str(device)=='cpu':
-            check_point = self.load_safetensor(model_path, map_location=torch.device('cpu'))
         else:
             check_point = self.load_safetensor(model_path)
         self.model.load_state_dict(check_point)

         self.model = ResnetEncoderDecoder(char_dict).to(device)
         self.model.eval()
         if str(device)=='cpu':
+            check_point = self.load_safetensor(model_path, map_location='cpu')
         else:
             check_point = self.load_safetensor(model_path)
         self.model.load_state_dict(check_point)