Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -257,7 +257,7 @@ def vos_tracking_video(video_state, interactive_state, mask_dropdown):
         template_mask[0][0]=1
         operation_log = [("Please add at least one mask to track by clicking the image in step2.","Error"), ("","")]
         # return video_output, video_state, interactive_state, operation_error
-    chunk_size = 5  # можно попробовать 3, если всё ещё вылетает
+    chunk_size = 5
     masks = []
     logits = []
     painted_images = []
@@ -266,15 +266,17 @@ def vos_tracking_video(video_state, interactive_state, mask_dropdown):
         end = min(start + chunk_size, len(following_frames))
         chunk_frames = following_frames[start:end]
 
+    # всегда используем одну и ту же template_mask
         chunk_masks, chunk_logits, chunk_painted = model.generator(
             images=chunk_frames,
-            template_mask=template_mask if start == 0 else chunk_masks[-1]
+            template_mask=template_mask
         )
 
         masks.extend(chunk_masks)
         logits.extend(chunk_logits)
         painted_images.extend(chunk_painted)
 
+
     model.cutie.clear_memory()
 
     if interactive_state["track_end_number"]: