Update app.py

app.py

@@ -52,7 +52,8 @@ def gemini_predict(prompt):
 
 
 MAX_SEED = np.iinfo(np.int32).max
-DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
+SAM_device = "cuda" # or "cpu"
+DEVICE = "cuda"
 
 
 ###GroundingDINO & SAM Setup
@@ -149,7 +150,6 @@ def refine_masks(masks: torch.BoolTensor, polygon_refinement: bool = False) -> L
 
 def get_alphacomp_mask(mask, image, random_color=True):
     annotated_frame_pil = Image.fromarray(image).convert("RGBA")
-    #mask_image_pil = Image.fromarray((mask_image.cpu().numpy() * 255).astype(np.uint8)).convert("RGBA")
     mask_image_pil = Image.fromarray(mask).convert("RGBA")
     
     return np.array(Image.alpha_composite(annotated_frame_pil, mask_image_pil))