killwithabass commited on
Commit
f106cd8
·
verified ·
1 Parent(s): add0661

add negative prompt

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -31,7 +31,7 @@ pipe.to("cuda")
31
  MAX_SEED = 2**32-1
32
 
33
  @spaces.GPU(duration=80)
34
- def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
35
  # Set random seed for reproducibility
36
  if randomize_seed:
37
  seed = random.randint(0, MAX_SEED)
@@ -52,6 +52,7 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
52
  # Generate image using the pipeline
53
  image = pipe(
54
  prompt=f"{prompt} {trigger_word}",
 
55
  num_inference_steps=steps,
56
  guidance_scale=cfg_scale,
57
  width=width,
@@ -108,6 +109,7 @@ with gr.Blocks() as app:
108
  with gr.Row():
109
  with gr.Column(scale=3):
110
  prompt = gr.TextArea(label="Prompt", placeholder="Type a prompt of max 77 characters", lines=3)
 
111
  generate_button = gr.Button("Generate")
112
  cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=example_cfg_scale)
113
  steps = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=example_steps)
@@ -125,7 +127,7 @@ with gr.Blocks() as app:
125
 
126
  generate_button.click(
127
  run_lora,
128
- inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale],
129
  outputs=[result, seed],
130
  )
131
 
 
31
  MAX_SEED = 2**32-1
32
 
33
  @spaces.GPU(duration=80)
34
+ def run_lora(prompt, negative_prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
35
  # Set random seed for reproducibility
36
  if randomize_seed:
37
  seed = random.randint(0, MAX_SEED)
 
52
  # Generate image using the pipeline
53
  image = pipe(
54
  prompt=f"{prompt} {trigger_word}",
55
+ negative_prompt,
56
  num_inference_steps=steps,
57
  guidance_scale=cfg_scale,
58
  width=width,
 
109
  with gr.Row():
110
  with gr.Column(scale=3):
111
  prompt = gr.TextArea(label="Prompt", placeholder="Type a prompt of max 77 characters", lines=3)
112
+ negative_prompt = gr.TextArea(label="Prompt", placeholder="Type a prompt of max 77 characters", lines=3)
113
  generate_button = gr.Button("Generate")
114
  cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=example_cfg_scale)
115
  steps = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=example_steps)
 
127
 
128
  generate_button.click(
129
  run_lora,
130
+ inputs=[prompt, negative_prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale],
131
  outputs=[result, seed],
132
  )
133