| --- |
| license: mit |
| pipeline_tag: text-classification |
| tags: |
| - model_hub_mixin |
| - pytorch_model_hub_mixin |
| language: |
| - en |
| metrics: |
| - accuracy |
| - precision |
| - recall |
| - f1 |
| base_model: |
| - google/vit-base-patch16-224 |
| library_name: timm |
| --- |
| |
| This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration: |
| - Code: https://huggingface.co/harriskr14/garbage_classifier |
| - Paper: [More Information Needed] |
| - Docs: [More Information Needed] |
| |
| # **How to use this model?** |
| |
| Write this following code on your file: |
| |
| **`app.py`** |
| ``` |
| def build_model(num_classes=10): |
| config = get_config() |
| model_name = 'vit_base_patch16_224' |
| model = timm.create_model(model_name, pretrained=True, num_classes=num_classes) |
| return model |
| |
| class MyModel(nn.Module, PyTorchModelHubMixin, repo_url="https://huggingface.co/harriskr14/garbage_classifier", pipeline_tag="text-classification", license="mit"): |
| def __init__(self, num_classes=10): |
| super(MyModel, self).__init__() |
| self.model = build_model(num_classes=num_classes) |
| |
| def forward(self, x): |
| return self.model(x) |
| |
| model = MyModel() |
| model.from_pretrained("harriskr14/garbage_classifier") |
| ``` |
