Dagfinn1962 commited on
Commit
a6795dd
1 Parent(s): b35bab7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -118,7 +118,7 @@ models = [
118
 
119
  current_model = models[0]
120
 
121
- text_gen = gr.Interface.load("spaces/daspartho/prompt-extend")
122
 
123
  models2 = []
124
  for model in models:
 
118
 
119
  current_model = models[0]
120
 
121
+ gpt2_pipe = pipeline('text-generation', model='https://huggingface.co/Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2')
122
 
123
  models2 = []
124
  for model in models: