Pendrokar commited on
Commit
2b5c8ca
·
1 Parent(s): 515b1f6

save_pretrained

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -13,6 +13,7 @@ from torchmoji.model_def import torchmoji_emojis
13
  from transformers import AutoModel, AutoTokenizer
14
  model_name = "Pendrokar/TorchMoji"
15
  model = AutoModel.from_pretrained(model_name, cache_dir="~/.cache/huggingface/hub/")
 
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  model_path = "~/.cache/huggingface/hub/TorchMoji/pytorch_model.bin"
18
  vocab_path = './' + model_name + "/vocabulary.json"
 
13
  from transformers import AutoModel, AutoTokenizer
14
  model_name = "Pendrokar/TorchMoji"
15
  model = AutoModel.from_pretrained(model_name, cache_dir="~/.cache/huggingface/hub/")
16
+ model.save_pretrained("~/.cache/huggingface/hub/TorchMoji/pytorch_model.bin")
17
  tokenizer = AutoTokenizer.from_pretrained(model_name)
18
  model_path = "~/.cache/huggingface/hub/TorchMoji/pytorch_model.bin"
19
  vocab_path = './' + model_name + "/vocabulary.json"