Hugging Face's logo

Spaces:
dipankardas011
/
gpt2-bot
Runtime error

App Files Community
dipankardas011 commited on
Commit
04b233c
·
unverified ·
1 Parent(s): a4fea27

due to performance issue switched to' gpt2-medium

Browse files

Signed-off-by: Dipankar Das <dipankardas0115@gmail.com>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from fastapi.responses import RedirectResponse
3
  import requests
4
 
5
  from transformers import pipeline, set_seed
6
- generator = pipeline('text-generation', model='gpt2-large')
7
  set_seed(42)
8
 
9
 
 
3
  import requests
4
 
5
  from transformers import pipeline, set_seed
6
+ generator = pipeline('text-generation', model='gpt2-medium')
7
  set_seed(42)
8
 
9