Fetching metadata from the HF Docker repository... 
Update Dockerfile
fd83457 verified - 1.52 kB initial commit
- 159 Bytes Update Dockerfile
- 257 Bytes initial commit
- 661 Bytes Create app.py
- 1.51 kB Update main.py
model.pt Detected Pickle imports (12)
- "torch._utils._rebuild_tensor_v2",
- "torch.nn.modules.container.Sequential",
- "__builtin__.set",
- "torch.FloatStorage",
- "torch.nn.modules.conv.Conv2d",
- "torch.nn.modules.pooling.MaxPool2d",
- "torch.nn.modules.pooling.AdaptiveAvgPool2d",
- "collections.OrderedDict",
- "torch.nn.modules.linear.Linear",
- "__main__.UNetClassifier",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.activation.ReLU"
How to fix it?
1.05 MB Upload model.pt - 1.05 MB Upload model_weights.pth
- 65 Bytes Update requirements.txt
