Fetching metadata from the HF Docker repository...
Create yolo_predictor.py
fc5324e verified - 1.62 kB Add NDVI model with LFS
4c_6c_regression.pt Detected Pickle imports (31)
- "ultralytics.nn.modules.block.Proto",
- "ultralytics.nn.modules.conv.Concat",
- "ultralytics.nn.modules.block.C3k",
- "collections.OrderedDict",
- "ultralytics.nn.modules.block.Attention",
- "torch.Size",
- "ultralytics.nn.modules.conv.Conv",
- "torch.LongStorage",
- "ultralytics.nn.modules.block.DFL",
- "torch.nn.modules.linear.Identity",
- "__builtin__.set",
- "torch._utils._rebuild_tensor_v2",
- "torch.HalfStorage",
- "torch.nn.modules.batchnorm.BatchNorm2d",
- "ultralytics.nn.modules.block.PSABlock",
- "torch.nn.modules.activation.SiLU",
- "ultralytics.nn.modules.block.C2PSA",
- "torch.FloatStorage",
- "ultralytics.nn.tasks.SegmentationModel",
- "torch.nn.modules.conv.Conv2d",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.conv.ConvTranspose2d",
- "ultralytics.nn.modules.conv.DWConv",
- "torch.nn.modules.container.ModuleList",
- "ultralytics.nn.modules.block.C3k2",
- "ultralytics.nn.modules.block.SPPF",
- "ultralytics.nn.modules.head.Segment",
- "ultralytics.nn.modules.block.Bottleneck",
- "torch.nn.modules.container.Sequential",
- "torch.nn.modules.pooling.MaxPool2d",
- "torch.nn.modules.upsampling.Upsample"
How to fix it?
6 MB Upload 4c_6c_regression.pt - 630 Bytes Update Dockerfile
- 193 Bytes initial commit
- 5.85 kB Update app.py
- 122 MB Upload ndvi_best_model.keras
- 2.81 kB ndvi_pred only
- 101 Bytes Update requirements.txt
- 5.27 kB Create yolo_predictor.py