modified app.py and Dockerfile

Dockerfile

@@ -14,5 +14,5 @@ COPY --chown=user . /home/user/app
 RUN wget -O output.zip -q https://huggingface.co/camenduru/Dynamic3DGaussians/resolve/main/output.zip?download=true && unzip output.zip && rm output.zip
 RUN cp -R /home/user/app/Dynamic3DGaussians /home/user/app/Dynamic3DGaussians_cpu
 RUN sed -i s/\.cuda\(\)\./\./ /home/user/app/Dynamic3DGaussians_cpu/*.py
-RUN sed -i s/, device="cuda"// /home/user/app/Dynamic3DGaussians_cpu/*.py
+RUN sed -i s/,\ device="cuda"// /home/user/app/Dynamic3DGaussians_cpu/*.py
 ENTRYPOINT ["python", "/home/user/app/app.py"]

app.py

@@ -1,8 +1,13 @@
 import torch
 import sys
 if torch.cuda.is_available():
-	sys.path.append("/home/user/app/Dynamic3DGaussians")
-	from Dynamic3DGaussians.visualize import visualize
+	try:
+		torch.ones(1, 1).cuda().float()
+		sys.path.append("/home/user/app/Dynamic3DGaussians")
+		from Dynamic3DGaussians.visualize import visualize
+	except:
+		sys.path.append("/home/user/app/Dynamic3DGaussians_cpu")
+		from Dynamic3DGaussians_cpu.visualize import visualize
 else:
 	sys.path.append("/home/user/app/Dynamic3DGaussians_cpu")
 	from Dynamic3DGaussians_cpu.visualize import visualize