diff --git a/deploy/pdserving/win/ocr_web_server.py b/deploy/pdserving/win/ocr_web_server.py index 1f1394fb72acebe58ef5b2a03bf27cd25838c536..9fc94901295d2470a4a16289b67daf0a6cbdc52a 100644 --- a/deploy/pdserving/win/ocr_web_server.py +++ b/deploy/pdserving/win/ocr_web_server.py @@ -42,7 +42,7 @@ class OCRService(WebService): self.det_client = LocalPredictor() if sys.argv[1] == 'gpu': self.det_client.load_model_config( - det_model_config, use_gpu=True, gpu_id=1) + det_model_config, use_gpu=True, gpu_id=0) elif sys.argv[1] == 'cpu': self.det_client.load_model_config(det_model_config) self.ocr_reader = OCRReader(