Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -257,8 +257,24 @@ 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
-    masks, logits, painted_images = model.generator(images=following_frames, template_mask=template_mask)
-    # clear GPU memory
+    chunk_size = 5  # можно попробовать 3, если всё ещё вылетает
+    masks = []
+    logits = []
+    painted_images = []
+
+    for start in range(0, len(following_frames), chunk_size):
+        end = min(start + chunk_size, len(following_frames))
+        chunk_frames = following_frames[start:end]
+
+        chunk_masks, chunk_logits, chunk_painted = model.generator(
+            images=chunk_frames,
+            template_mask=template_mask if start == 0 else chunk_masks[-1]
+        )
+
+        masks.extend(chunk_masks)
+        logits.extend(chunk_logits)
+        painted_images.extend(chunk_painted)
+
     model.cutie.clear_memory()
 
     if interactive_state["track_end_number"]: