Create push_to_hub.py

push_to_hub.py

@@ -0,0 +1,17 @@
+import torch
+import torch.nn as nn
+from huggingface_hub import PyTorchModelHubMixin
+from model import BERT
+
+# create model
+config = {"num_channels": 3, "hidden_size": 32, "num_classes": 10}
+model = BERT()
+
+# save locally
+model.save_pretrained("genre-pred")
+
+# push to the hub
+model.push_to_hub("genre-pred")
+
+# reload
+model = MyModel.from_pretrained("rishikasrinivas/genre-pred")