Robzy commited on
Commit
3dd836d
·
1 Parent(s): 8904a59

no local model saving

Browse files
Files changed (1) hide show
  1. model.py +3 -3
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="text-to-image",
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")