Update app.py
Browse files
app.py
CHANGED
|
@@ -89,17 +89,7 @@ def randomize_seed(seed, randomize):
|
|
| 89 |
return random.randint(0, MAX_SEED) if randomize else seed
|
| 90 |
|
| 91 |
@spaces.GPU
|
| 92 |
-
def create_image(
|
| 93 |
-
input_image,
|
| 94 |
-
prompt,
|
| 95 |
-
scale,
|
| 96 |
-
guidance_scale,
|
| 97 |
-
num_inference_steps,
|
| 98 |
-
seed,
|
| 99 |
-
style_mode,
|
| 100 |
-
negative_prompt="", # NEW (default empty = no change)
|
| 101 |
-
lora_scale=1.0 # NEW (default 1.0 = no change)
|
| 102 |
-
):
|
| 103 |
input_image = remove_bkg(input_image)
|
| 104 |
|
| 105 |
if style_mode == "Makoto Shinkai style":
|
|
@@ -114,7 +104,7 @@ def create_image(
|
|
| 114 |
generator = torch.manual_seed(seed)
|
| 115 |
common_args = dict(
|
| 116 |
prompt=prompt,
|
| 117 |
-
negative_prompt=negative_prompt, #
|
| 118 |
num_inference_steps=num_inference_steps,
|
| 119 |
guidance_scale=guidance_scale,
|
| 120 |
width=1024, height=1024,
|
|
@@ -124,12 +114,7 @@ def create_image(
|
|
| 124 |
)
|
| 125 |
|
| 126 |
if lora_path:
|
| 127 |
-
result = pipe.with_style_lora(
|
| 128 |
-
lora_file_path=lora_path,
|
| 129 |
-
trigger=trigger,
|
| 130 |
-
lora_scale=lora_scale, # NEW
|
| 131 |
-
**common_args
|
| 132 |
-
)
|
| 133 |
else:
|
| 134 |
result = pipe(**common_args)
|
| 135 |
return result.images
|
|
@@ -137,9 +122,9 @@ def create_image(
|
|
| 137 |
# --------------------------------------------
|
| 138 |
# UI definition (Gradio 5)
|
| 139 |
# --------------------------------------------
|
| 140 |
-
def generate_fn(image, prompt, scale, style, guidance, steps, seed, randomize, negative_prompt
|
| 141 |
seed = randomize_seed(seed, randomize)
|
| 142 |
-
return create_image(image, prompt, scale, guidance, steps, seed, style, negative_prompt
|
| 143 |
|
| 144 |
title = "🎨 InstantCharacter + One Piece LoRA"
|
| 145 |
description = (
|
|
@@ -159,16 +144,15 @@ demo = gr.Interface(
|
|
| 159 |
gr.Slider(5, 50, value=28, step=1, label="Inference Steps"),
|
| 160 |
gr.Slider(-1000000, 1000000, value=123456, step=1, label="Seed"),
|
| 161 |
gr.Checkbox(value=True, label="Randomize Seed"),
|
| 162 |
-
gr.Textbox(label="Negative Prompt",
|
| 163 |
-
gr.Slider(0.5, 1.5, value=1.0, step=0.01, label="LoRA Scale"), # NEW (default 1.0)
|
| 164 |
],
|
| 165 |
outputs=gr.Gallery(label="Generated Image"),
|
| 166 |
title=title,
|
| 167 |
description=description,
|
| 168 |
examples=[
|
| 169 |
-
["./assets/girl.jpg", f"A girl playing guitar, {ONEPIECE_TRIGGER}", 0.9, "One Piece style", 3.5, 28, 123, False, ""
|
| 170 |
-
["./assets/boy.jpg", f"A boy riding a bike, {ONEPIECE_TRIGGER}", 0.9, "One Piece style", 3.5, 28, 123, False, ""
|
| 171 |
-
]
|
| 172 |
)
|
| 173 |
|
| 174 |
demo.launch()
|
|
|
|
| 89 |
return random.randint(0, MAX_SEED) if randomize else seed
|
| 90 |
|
| 91 |
@spaces.GPU
|
| 92 |
+
def create_image(input_image, prompt, scale, guidance_scale, num_inference_steps, seed, style_mode, negative_prompt=""):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
input_image = remove_bkg(input_image)
|
| 94 |
|
| 95 |
if style_mode == "Makoto Shinkai style":
|
|
|
|
| 104 |
generator = torch.manual_seed(seed)
|
| 105 |
common_args = dict(
|
| 106 |
prompt=prompt,
|
| 107 |
+
negative_prompt=negative_prompt, # <-- ajouté
|
| 108 |
num_inference_steps=num_inference_steps,
|
| 109 |
guidance_scale=guidance_scale,
|
| 110 |
width=1024, height=1024,
|
|
|
|
| 114 |
)
|
| 115 |
|
| 116 |
if lora_path:
|
| 117 |
+
result = pipe.with_style_lora(lora_file_path=lora_path, trigger=trigger, **common_args)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 118 |
else:
|
| 119 |
result = pipe(**common_args)
|
| 120 |
return result.images
|
|
|
|
| 122 |
# --------------------------------------------
|
| 123 |
# UI definition (Gradio 5)
|
| 124 |
# --------------------------------------------
|
| 125 |
+
def generate_fn(image, prompt, scale, style, guidance, steps, seed, randomize, negative_prompt):
|
| 126 |
seed = randomize_seed(seed, randomize)
|
| 127 |
+
return create_image(image, prompt, scale, guidance, steps, seed, style, negative_prompt)
|
| 128 |
|
| 129 |
title = "🎨 InstantCharacter + One Piece LoRA"
|
| 130 |
description = (
|
|
|
|
| 144 |
gr.Slider(5, 50, value=28, step=1, label="Inference Steps"),
|
| 145 |
gr.Slider(-1000000, 1000000, value=123456, step=1, label="Seed"),
|
| 146 |
gr.Checkbox(value=True, label="Randomize Seed"),
|
| 147 |
+
gr.Textbox(label="Negative Prompt", placeholder="e.g. photorealistic, realistic skin, pores, hdr") # <-- ajouté
|
|
|
|
| 148 |
],
|
| 149 |
outputs=gr.Gallery(label="Generated Image"),
|
| 150 |
title=title,
|
| 151 |
description=description,
|
| 152 |
examples=[
|
| 153 |
+
["./assets/girl.jpg", f"A girl playing guitar, {ONEPIECE_TRIGGER}", 0.9, "One Piece style", 3.5, 28, 123, False, ""],
|
| 154 |
+
["./assets/boy.jpg", f"A boy riding a bike, {ONEPIECE_TRIGGER}", 0.9, "One Piece style", 3.5, 28, 123, False, ""]
|
| 155 |
+
]
|
| 156 |
)
|
| 157 |
|
| 158 |
demo.launch()
|