azaninello commited on
Commit
4211515
1 Parent(s): 3304da1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import GPT2Tokenizer, GPT2Model
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=120)
10
  general_result[0]["generated_text"]
11
 
12
 
 
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=200)
10
  general_result[0]["generated_text"]
11
 
12