Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
0982ce8
1
Parent(s):
3c7a85f
add
Browse files- app.py +5 -0
- requirements.txt +0 -1
app.py
CHANGED
@@ -1,4 +1,9 @@
|
|
1 |
import os
|
|
|
|
|
|
|
|
|
|
|
2 |
import torch
|
3 |
import torch.nn as nn
|
4 |
import gradio as gr
|
|
|
1 |
import os
|
2 |
+
# this is a HF Spaces specific hack, as
|
3 |
+
# (i) building pytorch3d with GPU support is a bit tricky here
|
4 |
+
# (ii) installing the wheel via requirements.txt breaks ZeroGPU
|
5 |
+
os.system("pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/download.html")
|
6 |
+
|
7 |
import torch
|
8 |
import torch.nn as nn
|
9 |
import gradio as gr
|
requirements.txt
CHANGED
@@ -21,4 +21,3 @@ rembg
|
|
21 |
onnxruntime
|
22 |
kiui
|
23 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
24 |
-
git+https://github.com/facebookresearch/pytorch3d.git@stable
|
|
|
21 |
onnxruntime
|
22 |
kiui
|
23 |
--extra-index-url https://download.pytorch.org/whl/cu118
|
|