haodongli commited on
Commit
73d2593
1 Parent(s): ddfdb2e

Update gradio_demo.py

Browse files
Files changed (1) hide show
  1. gradio_demo.py +1 -0
gradio_demo.py CHANGED
@@ -55,6 +55,7 @@ def main(prompt, init_prompt, negative_prompt, num_iter, CFG, seed):
55
  op.iterations = num_iter
56
  gp.guidance_scale = CFG
57
  gp.noise_seed = int(seed)
 
58
  if os.environ.get('QUEUE_1') != "True":
59
  os.environ['QUEUE_1'] = "True"
60
  lp.workspace = 'gradio_demo_1'
 
55
  op.iterations = num_iter
56
  gp.guidance_scale = CFG
57
  gp.noise_seed = int(seed)
58
+ print('==> User Prompt:', gp.text)
59
  if os.environ.get('QUEUE_1') != "True":
60
  os.environ['QUEUE_1'] = "True"
61
  lp.workspace = 'gradio_demo_1'