azaninello commited on
Commit
c813eac
·
1 Parent(s): 1dfbda9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ tokenizer = GPT2Tokenizer.from_pretrained(
8
  'LorenzoDeMattei/GePpeTto',
9
  )
10
 
11
- text_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('azaninello/GPT2-icc'),
12
  tokenizer='LorenzoDeMattei/GePpeTto',
13
  do_sample=True,
14
  max_length=200,
 
8
  'LorenzoDeMattei/GePpeTto',
9
  )
10
 
11
+ text_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('azaninello/GPT2-icc-new'),
12
  tokenizer='LorenzoDeMattei/GePpeTto',
13
  do_sample=True,
14
  max_length=200,