move

app.py

@@ -51,6 +51,7 @@ pipe = StableDiffusionXLFillPipeline.from_pretrained(
 pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 pipe.to("cuda")
 print(pipe)
+PREDICTOR = SAM2ImagePredictor.from_pretrained(SAM_MODEL, device=DEVICE)
 
 def load_default_pipeline():
     global pipe
@@ -67,7 +68,6 @@ def predict_masks(image, points):
     """Predict a single mask from the image based on selected points."""
     if not points:
         return image  # Return the original image if no points are selected
-    PREDICTOR = SAM2ImagePredictor.from_pretrained(SAM_MODEL, device=DEVICE)
 
     # Debugging: Print the structure of points
     print(f"Points structure: {points}")