Fetching metadata from the HF Docker repository...
Update requirements.txt
b2242cd
verified
-
Dataset
Delete Dataset/zero-indexed-files.txt
-
1.52 kB
initial commit
-
1.07 kB
Update LICENSE.txt
-
249 Bytes
Update README.md
-
3.59 kB
Update app.py
best.pt
Detected Pickle imports (26)
- "collections.OrderedDict",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "torch.nn.modules.linear.Linear",
- "transformers.models.vit.configuration_vit.ViTConfig",
- "transformers.models.vit.modeling_vit.ViTPooler",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "transformers.models.vit.modeling_vit.ViTModel",
- "torch.nn.modules.dropout.Dropout",
- "torch._utils._rebuild_parameter",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "src.classification.vit_model.ViTForImageClassification",
- "torch._C._nn.gelu",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "torch.nn.modules.activation.Tanh",
- "__builtin__.set",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "torch.FloatStorage",
- "torch.nn.modules.container.ModuleList",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "transformers.activations.GELUActivation",
- "transformers.models.vit.modeling_vit.ViTSdpaSelfAttention",
- "torch.nn.modules.conv.Conv2d",
- "transformers.models.vit.modeling_vit.ViTSdpaAttention"
How to fix it?
346 MB
Upload best.pt
-
215 MB
Upload classifier.h5
-
215 MB
Upload classify_model (5).h5
-
215 MB
Rename EcoIdentify_classification_model.h5 to classify_model.h5
-
73 Bytes
Update requirements.txt
-
821 Bytes
Update utils.py