Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -315,9 +315,9 @@ def inpaint_video(video_state, *_args):
         if i not in inpaint_mask_numbers:
             inpaint_masks[inpaint_masks == i] = 0
 
-    chunk_size = 30      # увеличим размер чанка
-    save_size = 20       # сколько сохраняем из каждого чанка (уникальное)
-    step = save_size     # шаг между чанками
+    chunk_size = 30
+    save_size = 20
+    step = save_size
     fixed_resize_ratio = 1.0
     fixed_dilate_radius = 8
     fixed_raft_iter = 20
@@ -326,6 +326,7 @@ def inpaint_video(video_state, *_args):
 
     total_len = len(frames)
     inpainted_all = []
+    saved_indices = set()
 
     for start in range(0, total_len, step):
         end = min(start + chunk_size, total_len)
@@ -345,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)
+            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(