Upload app.py with huggingface_hub

app.py

@@ -414,7 +414,7 @@ def replace_background(clip, prompt, relight, budget, steps, seed, keep_audio, r
 
     from h3_inpaint_crop import canvas_for_box, paste_back
 
-    progress(0.05, desc="Lifting the subject (alpha matte) ...")
+    progress(0.1, desc="Removing the background (LTX alpha matte) ...")
     green_path, matte_path = cached_matte(clip, int(seed))
     green = read_video(green_path)
     matte = read_video(matte_path)
@@ -451,7 +451,7 @@ def replace_background(clip, prompt, relight, budget, steps, seed, keep_audio, r
 
     references = [("image", ref_path)]
     full_prompt = "<Picture 1> " + prompt.strip()
-    progress(0.15, desc="Reading the prompt and reference ...")
+    progress(0.4, desc="Generating the new background (MiniMax-H3) ...")
     prompt_embeds, text_token_tags, metadata, _ = encode_remote(
         full_prompt, references, nearest_canvas_label(box[2], box[3]), n)
     num_frames = int(metadata["num_frames"])
@@ -463,7 +463,7 @@ def replace_background(clip, prompt, relight, budget, steps, seed, keep_audio, r
         "source_video": green, "mask": mask, "source_audio": waveform, "source_audio_sample_rate": sample_rate,
         "height": height, "width": width, "num_frames": num_frames, "turbo": True,
     }
-    progress(0.25, desc=f"Repainting the background at {width}x{height} ...")
+    progress(0.5, desc=f"Generating the new background (MiniMax-H3, {width}x{height}, ~1-2 min) ...")
     generated, audio, sampling_rate = _generate(request, steps, seed)
 
     progress(0.95, desc="Finishing ...")
@@ -505,6 +505,12 @@ EXAMPLES = [
      "a concert stage with dramatic blue and magenta spotlights and haze, dark background, cinematic", 0.0],
 ]
 
+def run_classic(clip, prompt, relight, budget, steps, seed, keep_audio, remove_only, progress=gr.Progress()):
+    """Wraps replace_background so gradio injects a progress tracker - the handler then reports its stages
+    (removing the background, then generating the new one) on the UI progress bar."""
+    return replace_background(clip, prompt, relight, budget, steps, seed, keep_audio, remove_only, progress=progress)
+
+
 with gr.Blocks(title="Video Replace Background") as demo:
     gr.Markdown(
         "# Video Replace Background\n"
@@ -538,7 +544,7 @@ with gr.Blocks(title="Video Replace Background") as demo:
 
     gr.Examples(examples=EXAMPLES, inputs=[clip, prompt, relight], label="Examples", cache_examples=False)
 
-    run.click(replace_background, [clip, prompt, relight, budget, steps, seed, keep_audio, remove_bg],
+    run.click(run_classic, [clip, prompt, relight, budget, steps, seed, keep_audio, remove_bg],
               [result, matte_view, mask_view], api_name="replace_background")