Hugging Face's logo

Spaces:
csuvikv
/
gpt-j-6b
Runtime error

App Files Community
csuvikv commited on
Commit
7748658
·
1 Parent(s): 3553dd8

neo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline, set_seed
3
 
4
 
5
- generator = pipeline('text-generation', model='EleutherAI/gpt-j-6B', return_full_text=False)
6
  set_seed(42)
7
 
8
 
 
2
  from transformers import pipeline, set_seed
3
 
4
 
5
+ generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B', return_full_text=False)
6
  set_seed(42)
7
 
8