Nad54 commited on
Commit
41a18bb
·
verified ·
1 Parent(s): a762e0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -25
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, # NEW
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, lora_scale):
141
  seed = randomize_seed(seed, randomize)
142
- return create_image(image, prompt, scale, guidance, steps, seed, style, negative_prompt, lora_scale)
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", value=""), # NEW (default empty)
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, "", 1.0],
170
- ["./assets/boy.jpg", f"A boy riding a bike, {ONEPIECE_TRIGGER}", 0.9, "One Piece style", 3.5, 28, 123, False, "", 1.0],
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()