diff --git a/api_server.py b/api_server.py index 18434f4..015c1b2 100644 --- a/api_server.py +++ b/api_server.py @@ -33,7 +33,7 @@ class IrImage(BaseModel): reference_pattern_path = '/home/nils/kinect_reference_cropped.png' reference_pattern = cv2.imread(reference_pattern_path) -model_path = "../train_log/models/latest.pth" +model_path = "train_log/models/latest.pth" device = torch.device('cuda:0') model = Model(max_disp=256, mixed_precision=False, test_mode=True)