Spaces:
Runtime error
Runtime error
Upload gradio_app.py
Browse files- 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
|