akshaytrikha commited on
Commit
ba07099
β€’
1 Parent(s): 505265f

rename model to gpt2-124M

Browse files
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
6
 
7
  # load pretrained + finetuned GPT2
8
- model = GPT2LMHeadModel.from_pretrained("./model")
9
  model = model.to(device)
10
 
11
  # create tokenizer
 
5
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
6
 
7
  # load pretrained + finetuned GPT2
8
+ model = GPT2LMHeadModel.from_pretrained("./model/gpt2-124M")
9
  model = model.to(device)
10
 
11
  # create tokenizer
huggingface_trump.ipynb CHANGED
The diff for this file is too large to render. See raw diff
 
model/.DS_Store ADDED
Binary file (6.15 kB). View file
 
model/{config.json β†’ gpt2-124M/config.json} RENAMED
File without changes
model/{pytorch_model.bin β†’ gpt2-124M/pytorch_model.bin} RENAMED
File without changes
model/{training_args.bin β†’ gpt2-124M/training_args.bin} RENAMED
File without changes