Hugging Face's logo

Spaces:
AIGuardians
/
complete_Story
Build error

App Files Community
g3casey commited on
Commit
8b5793d
·
1 Parent(s): 4f606e3

Change to x. model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import numpy as np
4
  import torch
5
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
6
 
7
- gpt2_model = GPT2LMHeadModel.from_pretrained("gpt2-large")
8
- gpt2_tokenizer = GPT2Tokenizer.from_pretrained("gpt2-large")
9
 
10
 
11
  seed = random.randint(0, 13)
 
4
  import torch
5
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
6
 
7
+ gpt2_model = GPT2LMHeadModel.from_pretrained("gpt2-xl")
8
+ gpt2_tokenizer = GPT2Tokenizer.from_pretrained("gpt2-xl")
9
 
10
 
11
  seed = random.randint(0, 13)