Gustavosta commited on
Commit
9a8b416
1 Parent(s): f161ee4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from transformers import pipeline, set_seed
2
  import gradio as grad, random, re
3
 
4
 
5
- gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Dalle', tokenizer='gpt2')
6
  with open("ideas.txt", "r") as f:
7
  line = f.readlines()
8
 
 
2
  import gradio as grad, random, re
3
 
4
 
5
+ gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Dalle', tokenizer='gpt2', use_auth_token=True)
6
  with open("ideas.txt", "r") as f:
7
  line = f.readlines()
8