amazonaws-sp commited on
Commit
3c9832d
1 Parent(s): fab9cdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ def generate(
90
  negative_prompt_2 = None # type: ignore
91
 
92
  if not use_img2img:
93
- pipe = pipe(
94
  prompt=prompt,
95
  negative_prompt=negative_prompt,
96
  prompt_2=prompt_2,
 
90
  negative_prompt_2 = None # type: ignore
91
 
92
  if not use_img2img:
93
+ return pipe(
94
  prompt=prompt,
95
  negative_prompt=negative_prompt,
96
  prompt_2=prompt_2,