nastasiasnk commited on
Commit
5965e2a
·
verified ·
1 Parent(s): 9c084a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ from transformers import pipeline
20
  #generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B', device=device)
21
 
22
 
23
- generator = pipeline("text-generation", model="gpt-3")
24
 
25
 
26
  st.title("text class")
 
20
  #generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B', device=device)
21
 
22
 
23
+ generator = pipeline("text-generation", model="gpt-2")
24
 
25
 
26
  st.title("text class")