Spaces:
Paused
Paused
CrazyT commited on
Commit ·
47bf5f3
1
Parent(s): abe494a
should run now
Browse files- Dockerfile +4 -1
- app.py +9 -4
- requirements1.txt +6 -0
- requirements2.txt +1 -0
- runLocalDocker.bat +2 -0
Dockerfile
CHANGED
|
@@ -9,5 +9,8 @@ WORKDIR /home/user/app
|
|
| 9 |
#RUN --mount=target=/root/packages.txt,source=packages.txt apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
|
| 10 |
#Restoring cache
|
| 11 |
RUN --mount=target=pre-requirements.txt,source=pre-requirements.txt pip install --no-cache-dir -r pre-requirements.txt
|
| 12 |
-
|
|
|
|
|
|
|
|
|
|
| 13 |
ENTRYPOINT ["python", "app.py"]
|
|
|
|
| 9 |
#RUN --mount=target=/root/packages.txt,source=packages.txt apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
|
| 10 |
#Restoring cache
|
| 11 |
RUN --mount=target=pre-requirements.txt,source=pre-requirements.txt pip install --no-cache-dir -r pre-requirements.txt
|
| 12 |
+
ENV TORCH_CUDA_ARCH_LIST=Turing
|
| 13 |
+
RUN --mount=target=requirements1.txt,source=requirements1.txt pip install --no-cache-dir -r requirements1.txt
|
| 14 |
+
RUN --mount=target=requirements2.txt,source=requirements2.txt pip install --no-cache-dir -r requirements2.txt
|
| 15 |
+
RUN pip install gradio
|
| 16 |
ENTRYPOINT ["python", "app.py"]
|
app.py
CHANGED
|
@@ -1,7 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import gradio as gr
|
|
|
|
| 2 |
|
| 3 |
-
def
|
| 4 |
-
|
|
|
|
| 5 |
|
| 6 |
-
iface = gr.Interface(fn=
|
| 7 |
-
iface.launch()
|
|
|
|
| 1 |
+
import sys
|
| 2 |
+
sys.path.append("Dynamic3DGaussians")
|
| 3 |
+
|
| 4 |
import gradio as gr
|
| 5 |
+
from Dynamic3DGaussians.visualize import visualize
|
| 6 |
|
| 7 |
+
def basketball(name):
|
| 8 |
+
visualize("basketball", "pretrained")
|
| 9 |
+
return "Hello " + name + "!!!"
|
| 10 |
|
| 11 |
+
iface = gr.Interface(fn=basketball, inputs=None, outputs="text")
|
| 12 |
+
iface.launch(server_name="0.0.0.0")
|
requirements1.txt
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
pip==22.3.1
|
| 2 |
+
torchaudio==0.12.1
|
| 3 |
+
torchvision==0.13.1
|
| 4 |
+
tqdm==4.65.0
|
| 5 |
+
Pillow==9.4.0
|
| 6 |
+
open3d==0.16.0
|
requirements2.txt
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
diff-gaussian-rasterization @ git+https://github.com/JonathonLuiten/diff-gaussian-rasterization-w-depth.git@cb65e4b86bc3bd8ed42174b72a62e8d3a3a71110
|
runLocalDocker.bat
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
docker run --gpus=all --entrypoint "python" -v .:/home/user/app -p 7860:7860 --rm -ti dynamic3dgaussians /usr/local/bin/gradio app.py
|
| 2 |
+
pause
|