harriskr14
/

garbage_classifier

Text Classification

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions

harriskr14 commited on Mar 8

Commit

2b763a3

·

verified ·

1 Parent(s): 7890b96

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -35,7 +35,7 @@ def build_model(num_classes=10):
 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(DeployModel, self).__init__()
         self.model = build_model(num_classes=num_classes)
     def forward(self, x):

 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):