Spaces:
Build error
Build error
Vo Minh Vu
commited on
Commit
·
1e0f082
1
Parent(s):
f82011f
reinstall torchmcubes
Browse files- .DS_Store +0 -0
- app.py +7 -0
- wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl +3 -0
.DS_Store
ADDED
|
Binary file (6.15 kB). View file
|
|
|
app.py
CHANGED
|
@@ -21,6 +21,13 @@ from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orienta
|
|
| 21 |
# ------------------------------------------------------------
|
| 22 |
# 1. Model & utils initialization (runs at startup)
|
| 23 |
# ------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
# device
|
| 26 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 21 |
# ------------------------------------------------------------
|
| 22 |
# 1. Model & utils initialization (runs at startup)
|
| 23 |
# ------------------------------------------------------------
|
| 24 |
+
# Install any local wheels (if needed)
|
| 25 |
+
subprocess.run(
|
| 26 |
+
shlex.split(
|
| 27 |
+
"pip install wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl"
|
| 28 |
+
),
|
| 29 |
+
check=False,
|
| 30 |
+
)
|
| 31 |
|
| 32 |
# device
|
| 33 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4af160ba1274e2205d3529a7b82efdb6946c2158a78e19631ed840301055b8d6
|
| 3 |
+
size 5824388
|