no local model saving
Browse files
model.py
CHANGED
@@ -8,7 +8,7 @@ class MyModel(
|
|
8 |
PyTorchModelHubMixin,
|
9 |
# optionally, you can add metadata which gets pushed to the model card
|
10 |
repo_url="https://huggingface.co/Robzy/random-genre",
|
11 |
-
pipeline_tag="
|
12 |
license="mit",
|
13 |
):
|
14 |
def __init__(self, num_channels: int, hidden_size: int, num_classes: int):
|
@@ -24,6 +24,6 @@ config = {"num_channels": 3, "hidden_size": 32, "num_classes": 10}
|
|
24 |
model = MyModel(**config)
|
25 |
|
26 |
# Save the model locally
|
27 |
-
model_save_path = "trial-model"
|
28 |
-
model.save_pretrained(model_save_path)
|
29 |
model.push_to_hub("Robzy/random-genre")
|
|
|
8 |
PyTorchModelHubMixin,
|
9 |
# optionally, you can add metadata which gets pushed to the model card
|
10 |
repo_url="https://huggingface.co/Robzy/random-genre",
|
11 |
+
pipeline_tag="audio-classification",
|
12 |
license="mit",
|
13 |
):
|
14 |
def __init__(self, num_channels: int, hidden_size: int, num_classes: int):
|
|
|
24 |
model = MyModel(**config)
|
25 |
|
26 |
# Save the model locally
|
27 |
+
# model_save_path = "trial-model"
|
28 |
+
# model.save_pretrained(model_save_path)
|
29 |
model.push_to_hub("Robzy/random-genre")
|