Update web-demos/hugging_face/track_anything.py

web-demos/hugging_face/track_anything.py

@@ -21,21 +21,18 @@ class TrackingAnything():
         masks = []
         logits = []
         painted_images = []
-        alpha_visuals = []
         for i in tqdm(range(len(images)), desc="Tracking image"):
             if i==0:           
-                mask, logit, painted_image, alpha_visual = self.cutie.track(images[i], template_mask)
-                alpha_visuals.append(alpha_visual)
+                mask, logit, painted_image = self.cutie.track(images[i], template_mask)
                 masks.append(mask)
                 logits.append(logit)
                 painted_images.append(painted_image)
             else:
-                mask, logit, painted_image, alpha_visual = self.cutie.track(images[i])
+                mask, logit, painted_image = self.cutie.track(images[i])
                 masks.append(mask)
                 logits.append(logit)
-                painted_images.append(painted_image)
-                alpha_visuals.append(alpha_visual)
-        return masks, logits, painted_images, alpha_visuals
+                painted_images.append(painted_image)  
+        return masks, logits, painted_images