Upload app.py with huggingface_hub

app.py

@@ -418,11 +418,19 @@ def replace_background(clip, prompt, relight, budget, steps, seed, keep_audio, p
 
     progress(0.95, desc="Finishing ...")
     raw = np.stack([np.asarray(f) for f in generated])
+    gen = np.stack([np.asarray(Image.fromarray(f).resize((clip_w, clip_h), Image.Resampling.LANCZOS)) for f in raw])
     if float(relight) > 0:
         # keep H3's own frame so the relight it applied to the subject survives
-        frames = np.stack([np.asarray(Image.fromarray(f).resize((clip_w, clip_h))) for f in raw])
+        frames = gen
     else:
-        frames = paste_back(green, raw, (0, 0, clip_h, clip_w), mask=(mask > 0.5).astype(np.float32), feather=8)
+        # Full-frame composite: restore the exact subject pixels, feathering ONLY the subject boundary. paste_back's
+        # border ramp fades to the plate at the frame edges, which here is the green screen - so it is not used; the
+        # background (incl. the frame edge) is taken fully from the generated frame.
+        from h3_inpaint_crop import _feather_mask
+
+        weight = _feather_mask((mask > 0.5).astype(np.float32), 8)[..., None]
+        frames = (green.astype(np.float32) * (1 - weight) + gen.astype(np.float32) * weight)
+        frames = frames.round().clip(0, 255).astype(np.uint8)
 
     from diffusers.utils import encode_video
     import torch