Update app.py

app.py

@@ -125,7 +125,6 @@ class CustomSAMWrapper:
             print(f"SAM prediction error: {e}")
             return np.zeros((image.height, image.width), dtype=bool), 0.0
 
-sam_wrapper = CustomSAMWrapper(SAM_CHECKPOINT, device=DEVICE)
 
 
 def parse_custom_format(content: str):
@@ -234,6 +233,7 @@ def visualize_masks_on_image(
 @spaces.GPU
 @torch.no_grad()
 def run_pipeline(image: PILImage.Image, prompt: str):
+    sam_wrapper = CustomSAMWrapper(SAM_CHECKPOINT, device=DEVICE)
     if not model or not processor:
         return "Models not loaded. Please check logs.", None