Fetching metadata from the HF Docker repository...
config added
2ea938d - .streamlit config added
- 1.48 kB initial commit
- 270 Bytes initial commit
- 882 Bytes Update app.py
- 3.6 kB Update back.py
distilbert1.bin Detected Pickle imports (21)
- "torch._utils._rebuild_tensor_v2",
- "__builtin__.set",
- "transformers.models.distilbert.modeling_distilbert.FFN",
- "torch.nn.modules.sparse.Embedding",
- "torch.nn.modules.container.ModuleList",
- "transformers.activations.GELUActivation",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.functional.gelu",
- "torch.FloatStorage",
- "transformers.models.distilbert.modeling_distilbert.DistilBertModel",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.Linear",
- "transformers.models.distilbert.modeling_distilbert.MultiHeadSelfAttention",
- "torch.LongStorage",
- "transformers.models.distilbert.modeling_distilbert.Transformer",
- "transformers.models.distilbert.modeling_distilbert.Embeddings",
- "transformers.models.distilbert.modeling_distilbert.TransformerBlock",
- "collections.OrderedDict",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.distilbert.configuration_distilbert.DistilBertConfig",
- "__main__.DistillBERTClass"
How to fix it?
267 MB models - 787 Bytes Create model.py
- 28 Bytes Update requirements.txt
- 213 kB models