Revert

app_endframe.py

@@ -665,50 +665,6 @@ def worker(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, bat
 def get_duration(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
     return total_second_length * 60
 
-@spaces.GPU(duration=get_duration)
-def process_example(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
-    global stream, high_vram
-    # 20250506 pftq: Updated assertion for video input
-    assert input_video is not None, 'No input video!'
-
-    yield None
-
-    # 20250507 pftq: Even the H100 needs offloading if the video dimensions are 720p or higher
-    if high_vram and (no_resize or resolution>640):
-        print("Disabling high vram mode due to no resize and/or potentially higher resolution...")
-        high_vram = False
-        vae.enable_slicing()
-        vae.enable_tiling()
-        DynamicSwapInstaller.install_model(transformer, device=gpu)
-        DynamicSwapInstaller.install_model(text_encoder, device=gpu)
-        
-    # 20250508 pftq: automatically set distilled cfg to 1 if cfg is used
-    if cfg > 1:
-        gs = 1
-
-    stream = AsyncStream()
-
-    # 20250506 pftq: Pass num_clean_frames, vae_batch, etc
-    async_run(worker, input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch)
-
-    output_filename = None
-
-    while True:
-        flag, data = stream.output_queue.next()
-
-        if flag == 'file':
-            output_filename = data
-            yield output_filename
-
-        if flag == 'progress':
-            preview, desc, html = data
-            #yield gr.update(), gr.update(visible=True, value=preview), desc, html, gr.update(interactive=False), gr.update(interactive=True)
-            yield output_filename
-
-        if flag == 'end':
-            yield output_filename
-            break
-
 @spaces.GPU(duration=get_duration)
 def process(input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
     global stream, high_vram
@@ -858,9 +814,9 @@ with block:
                 ],
             ],
         run_on_click = True,
-        fn = process_example,
+        fn = process,
 	    inputs = [input_video, end_frame, end_frame_weight, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch],
-	    outputs = [result_video],
+	    outputs = [result_video, preview_image, progress_desc, progress_bar, start_button, end_button],
         cache_examples = True,
     )
     gr.HTML("""