Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -316,7 +316,7 @@ def inpaint_video(video_state, *_args):
             inpaint_masks[inpaint_masks == i] = 0
 
     chunk_size = 30
-    save_size = 20
+    save_size = 22
     step = save_size
     fixed_resize_ratio = 1.0
     fixed_dilate_radius = 8
@@ -326,20 +326,15 @@ def inpaint_video(video_state, *_args):
 
     total_len = len(frames)
     inpainted_all = []
-    last_chunk_result = None
+    saved_indices = set()
 
     for start in range(0, total_len, step):
         end = min(start + chunk_size, total_len)
         chunk_frames = frames[start:end]
         chunk_masks = inpaint_masks[start:end]
 
-        if end - start < 2 and last_chunk_result is not None:
-            print(f"Skipping tiny chunk {start}:{end}, copying from previous.")
-            remaining = end - start
-            inpainted_all.extend(last_chunk_result[-remaining:])
-            continue
-
         print(f"Inpainting chunk {start}:{end}")
+
         chunk_result = model.baseinpainter.inpaint(
             chunk_frames,
             chunk_masks,
@@ -351,15 +346,29 @@ def inpaint_video(video_state, *_args):
             ref_stride=fixed_ref_stride
         )
 
+        chunk_len = end - start
+
+        # Выбираем уникальные индексы для сохранения
         if start == 0:
-            chunk_to_save = chunk_result[:save_size]
+            to_save = list(range(min(save_size, chunk_len)))
         elif end == total_len:
-            chunk_to_save = chunk_result[(chunk_size - (end - start)):]
+            to_save = list(range(chunk_len - (total_len - start), chunk_len))
         else:
-            chunk_to_save = chunk_result[:save_size]
-
-        inpainted_all.extend(chunk_to_save)
-        last_chunk_result = chunk_result
+            to_save = list(range(min(save_size, chunk_len)))
+
+        for i in to_save:
+            absolute_index = start + i
+            if absolute_index not in saved_indices:
+                inpainted_all.append(chunk_result[i])
+                saved_indices.add(absolute_index)
+
+    # 🧠 Убедимся, что длина совпадает
+    if len(inpainted_all) < total_len:
+        last_frame = inpainted_all[-1]
+        for _ in range(total_len - len(inpainted_all)):
+            inpainted_all.append(last_frame)
+    elif len(inpainted_all) > total_len:
+        inpainted_all = inpainted_all[:total_len]
 
     output_path = "./result/inpaint/{}".format(video_state["video_name"])
     video_output = generate_video_from_frames(
@@ -370,6 +379,7 @@ def inpaint_video(video_state, *_args):
 
     return video_output, operation_log, operation_log
 
+
 # generate video after vos inference
 def generate_video_from_frames(frames, output_path, fps=30, bitrate=None):
     """