Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import io
 import numpy as np
 import torch
@@ -13,10 +14,17 @@ from transformers import AutoModelForImageSegmentation, pipeline
 # Set device (GPU if available, else CPU)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+# Retrieve Hugging Face token from environment variable
+hf_token = os.environ.get("HF_ACCESS_TOKEN")
+if hf_token is None:
+    print("Warning: HF_ACCESS_TOKEN is not set. Access to gated models may fail.")
+
 # Load the segmentation model (RMBG-2.0)
+# Ensure you have access to this gated repo and that your token is valid.
 segmentation_model = AutoModelForImageSegmentation.from_pretrained(
     'briaai/RMBG-2.0',
-    trust_remote_code=True
+    trust_remote_code=True,
+    use_auth_token=hf_token
 )
 segmentation_model.to(device)
 segmentation_model.eval()