Spaces:
Build error
Build error
Vo Minh Vu
commited on
Commit
·
f82011f
1
Parent(s):
bb92587
remove wheel
Browse files- app.py +0 -7
- requirements.txt +1 -0
- wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl +0 -3
app.py
CHANGED
|
@@ -21,13 +21,6 @@ from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orienta
|
|
| 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"
|
|
|
|
| 21 |
# ------------------------------------------------------------
|
| 22 |
# 1. Model & utils initialization (runs at startup)
|
| 23 |
# ------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
# device
|
| 26 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
requirements.txt
CHANGED
|
@@ -8,6 +8,7 @@ rembg
|
|
| 8 |
huggingface-hub
|
| 9 |
gradio
|
| 10 |
onnxruntime
|
|
|
|
| 11 |
|
| 12 |
fastapi
|
| 13 |
uvicorn[standard]
|
|
|
|
| 8 |
huggingface-hub
|
| 9 |
gradio
|
| 10 |
onnxruntime
|
| 11 |
+
torchmcubes
|
| 12 |
|
| 13 |
fastapi
|
| 14 |
uvicorn[standard]
|
wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:4af160ba1274e2205d3529a7b82efdb6946c2158a78e19631ed840301055b8d6
|
| 3 |
-
size 5824388
|
|
|
|
|
|
|
|
|
|
|
|