Spaces:
Running
on
Zero
Running
on
Zero
add negative prompt
Browse files
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 |
|