File size: 312 Bytes
86341ce f40c947 86341ce | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | import torch
import torch.nn as nn
from huggingface_hub import PyTorchModelHubMixin
from model import BERT
# create model
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") |