Fetching metadata from the HF Docker repository... 
Update requirements.txt
721c37c - 1.17 kB initial commit
- 43 Bytes Add model and demo
- 914 Bytes initial commit
- 3.34 kB Update app.py
- 44 kB Add model and demo
model-final.pth Detected Pickle imports (14)
- "torch.nn.modules.batchnorm.BatchNorm2d",
- "torch.nn.modules.activation.ReLU",
- "__builtin__.set",
- "__main__.ColorizationNet",
- "torch._utils._rebuild_parameter",
- "torch.LongStorage",
- "torch.nn.modules.upsampling.Upsample",
- "torchvision.models.resnet.BasicBlock",
- "torch.FloatStorage",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.pooling.MaxPool2d",
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.conv.Conv2d",
- "collections.OrderedDict"
How to fix it?
3.87 MB Add model and demo - 2.58 kB Add model and demo
- 252 kB Add model and demo
- 204 Bytes Update requirements.txt
