John6666 commited on
Commit
8464784
·
verified ·
1 Parent(s): dd5b16a

Upload gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +2 -0
gradio_app.py CHANGED
@@ -44,6 +44,7 @@ COND_FOVY = 0.591627
44
  BACKGROUND_COLOR = [0.5, 0.5, 0.5]
45
  OUTPUT_DIR = "output"
46
  os.makedirs(OUTPUT_DIR, exist_ok=True)
 
47
 
48
  # Initialize models
49
  device = spar3d_utils.get_device()
@@ -52,6 +53,7 @@ spar3d_model = SPAR3D.from_pretrained(
52
  "stabilityai/stable-point-aware-3d",
53
  config_name="config.yaml",
54
  weight_name="model.safetensors"
 
55
  ).eval().to(device)
56
 
57
  # Initialize camera parameters
 
44
  BACKGROUND_COLOR = [0.5, 0.5, 0.5]
45
  OUTPUT_DIR = "output"
46
  os.makedirs(OUTPUT_DIR, exist_ok=True)
47
+ HF_TOKEN = os.getenv("HF_TOKEN", None)
48
 
49
  # Initialize models
50
  device = spar3d_utils.get_device()
 
53
  "stabilityai/stable-point-aware-3d",
54
  config_name="config.yaml",
55
  weight_name="model.safetensors"
56
+ token=HF_TOKEN
57
  ).eval().to(device)
58
 
59
  # Initialize camera parameters