m-ric HF staff commited on
Commit
8a3de17
1 Parent(s): 46edcf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -113,8 +113,8 @@ STYLE = """
113
  from transformers import GPT2Tokenizer, AutoModelForCausalLM
114
  import numpy as np
115
 
116
- tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
117
- model = AutoModelForCausalLM.from_pretrained("gpt2")
118
  tokenizer.pad_token_id = tokenizer.eos_token_id
119
  pritn("Loading finished.")
120
  def generate_html(token, node):
 
113
  from transformers import GPT2Tokenizer, AutoModelForCausalLM
114
  import numpy as np
115
 
116
+ tokenizer = GPT2Tokenizer.from_pretrained("openai-community/gpt2")
117
+ model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
118
  tokenizer.pad_token_id = tokenizer.eos_token_id
119
  pritn("Loading finished.")
120
  def generate_html(token, node):