KevinQu7 commited on
Commit
dbbe9ee
·
1 Parent(s): 67b98e5
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -468,8 +468,9 @@ def main():
468
 
469
  if "HF_TOKEN_LOGIN" in os.environ:
470
  login(token=os.environ["HF_TOKEN_LOGIN"])
471
-
472
- pipe = MarigoldNormalsPipeline.from_pretrained(CHECKPOINT)
 
473
  try:
474
  import xformers
475
 
 
468
 
469
  if "HF_TOKEN_LOGIN" in os.environ:
470
  login(token=os.environ["HF_TOKEN_LOGIN"])
471
+
472
+ auth_token = os.environ.get("KEV_TOKEN")
473
+ pipe = MarigoldNormalsPipeline.from_pretrained(CHECKPOINT,use_auth_token=auth_token)
474
  try:
475
  import xformers
476