yeq6x commited on
Commit
3fa1ded
·
1 Parent(s): 5540810

力画像を入力画像と生成画像のアルファブレンド(入力30%、生成70%)にしました。

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -152,6 +152,16 @@ def infer(
152
  num_images_per_prompt=1,
153
  ).images
154
 
 
 
 
 
 
 
 
 
 
 
155
  # Return first result image and seed
156
  return result_images[0] if result_images else None, seed
157
 
 
152
  num_images_per_prompt=1,
153
  ).images
154
 
155
+ # Alpha blend: input (0.3) + generated (0.7)
156
+ if result_images and pil_image is not None:
157
+ generated_image = result_images[0]
158
+ # Resize input image to match generated image size if different
159
+ if pil_image.size != generated_image.size:
160
+ pil_image = pil_image.resize(generated_image.size, Image.Resampling.LANCZOS)
161
+ # Blend: 0.3 * input + 0.7 * generated
162
+ blended_image = Image.blend(pil_image, generated_image, alpha=0.7)
163
+ return blended_image, seed
164
+
165
  # Return first result image and seed
166
  return result_images[0] if result_images else None, seed
167