youa commited on
Commit
734df1f
·
1 Parent(s): dd3ab19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from transformers import AutoModelForCausalLM, AutoModelWithLMHead, AutoTokenize
4
  from transformers import GPT2Tokenizer, GPT2Model
5
 
6
 
7
- general_model = AutoModelForCausalLM.from_pretrained('gpt2')
8
- general_generator = pipeline("text-generation", model=general_model, tokenizer="gpt2")
9
  general_result = general_generator("Today is ", max_length=1500)
10
  general_result[0]["generated_text"]
11
 
 
4
  from transformers import GPT2Tokenizer, GPT2Model
5
 
6
 
7
+ general_model = AutoModelForCausalLM.from_pretrained('youa/gpt2')
8
+ general_generator = pipeline("text-generation", model=general_model, tokenizer="youa/gpt2")
9
  general_result = general_generator("Today is ", max_length=1500)
10
  general_result[0]["generated_text"]
11