Spaces:
Sleeping
Sleeping
fix env
Browse files
app.py
CHANGED
@@ -16,6 +16,7 @@ from huggingface_hub import snapshot_download
|
|
16 |
from model.model import MeshTransformer
|
17 |
from utils import apply_normalize, joint_filter, sample_pc
|
18 |
|
|
|
19 |
|
20 |
CONFIG_PATH = 'config/BPT-open-8k-8-16.yaml'
|
21 |
with open(CONFIG_PATH, "r") as f:
|
|
|
16 |
from model.model import MeshTransformer
|
17 |
from utils import apply_normalize, joint_filter, sample_pc
|
18 |
|
19 |
+
os.environ["CUDA_HOME"] = "/usr/local/cuda"
|
20 |
|
21 |
CONFIG_PATH = 'config/BPT-open-8k-8-16.yaml'
|
22 |
with open(CONFIG_PATH, "r") as f:
|