code cleanup

colorizers/siggraph17.py

@@ -164,6 +164,9 @@ def siggraph17(pretrained=True):
     if(pretrained):
         import torch.utils.model_zoo as model_zoo
         # model.load_state_dict(torch.load('model.pth'))
+        # local_path = 'model.pth'
+        # state_dict = torch.load(local_path, map_location=torch.device('cpu'))
+        # model.load_state_dict(state_dict)
         model.load_state_dict(model_zoo.load_url('https://colorizers.s3.us-east-2.amazonaws.com/siggraph17-df00044c.pth',map_location='cpu',check_hash=True))
     return model
 

model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:df00044c0a4d7c3edcecf6f75437ce346a66e7a42612d9b968e1a7e17dbc6f66
+size 136787426