Update app.py

app.py

@@ -15,6 +15,14 @@ num_frames, width, height = 49, 832, 480
 gpu_id = 0
 device = f'cuda:{gpu_id}'
 
+from huggingface_hub import login
+import os
+# 使用环境变量中的 token 登录
+token = os.environ.get("HF_TOKEN")
+if token:
+    login(token)
+
+
 rmbg_model = AutoModelForImageSegmentation.from_pretrained('briaai/RMBG-2.0', trust_remote_code=True)
 torch.set_float32_matmul_precision(['high', 'highest'][0])
 rmbg_model.to(device)