Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -326,15 +326,20 @@ def inpaint_video(video_state, *_args):
 
     total_len = len(frames)
     inpainted_all = []
-    saved_indices = set()
+    last_chunk_result = None
 
     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]
 
-        print(f"Inpainting chunk {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,
@@ -346,29 +351,15 @@ def inpaint_video(video_state, *_args):
             ref_stride=fixed_ref_stride
         )
 
-        chunk_len = end - start
-
-        # Выбираем уникальные индексы для сохранения
         if start == 0:
-            to_save = list(range(min(save_size, chunk_len)))
+            chunk_to_save = chunk_result[:save_size]
         elif end == total_len:
-            to_save = list(range(chunk_len - (total_len - start), chunk_len))
+            chunk_to_save = chunk_result[(chunk_size - (end - start)):]
         else:
-            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]
+            chunk_to_save = chunk_result[:save_size]
+
+        inpainted_all.extend(chunk_to_save)
+        last_chunk_result = chunk_result
 
     output_path = "./result/inpaint/{}".format(video_state["video_name"])
     video_output = generate_video_from_frames(