CrazyT commited on
Commit
47bf5f3
·
1 Parent(s): abe494a

should run now

Browse files
Files changed (5) hide show
  1. Dockerfile +4 -1
  2. app.py +9 -4
  3. requirements1.txt +6 -0
  4. requirements2.txt +1 -0
  5. 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
- RUN --mount=target=requirements.txt,source=requirements.txt pip install --no-cache-dir -r requirements.txt
 
 
 
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 greet(name):
4
- return "Hello " + name + "!!"
 
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
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