update to use specific gpu requirements - can't do cpu

.gitignore

@@ -1 +1,2 @@
 .venv
+results/*

pyproject.toml

@@ -2,4 +2,42 @@
 name = "gsplat2d"
 version = "0.1.0"
 requires-python = ">=3.10"
-dependencies = []
+dependencies = [
+    "gradio>=5.20.1",
+    "gsplat>=1.4",
+    "imageio[ffmpeg]>=2.37.0",
+    "matplotlib>=3.10.1",
+    "nerfview==0.0.3",
+    "numpy<2.0.0",
+    "opencv-python>=4.11.0.86",
+    "pillow>=11.1.0",
+    "pycolmap",
+    "pyyaml>=6.0.2",
+    "scikit-learn>=1.6.1",
+    "tensorly>=0.9.0",
+    "torch>=2.6.0",
+    "torchmetrics[image]>=1.6.2",
+    "torchvision>=0.21.0",
+    "tqdm>=4.67.1",
+    "tyro>=0.8.8",
+    "viser>=0.2.23",
+]
+
+[tool.uv.sources]
+pycolmap = { git = "https://github.com/rmbrualla/pycolmap", rev = "cc7ea4b7301720ac29287dbe450952511b32125e" }
+torch = [
+  { index = "pytorch-cu124", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+gsplat = [
+  { index = "gsplat-pt20cu124", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
+]
+
+[[tool.uv.index]]
+name = "pytorch-cu124"
+url = "https://download.pytorch.org/whl/cu124"
+explicit = true
+
+[[tool.uv.index]]
+name = "gsplat-pt20cu124"
+url = "https://docs.gsplat.studio/whl/pt24cu124"
+explicit = true

requirements.txt

@@ -1,4 +1,6 @@
-torch --index-url https://download.pytorch.org/whl/cpu
+--index-url https://download.pytorch.org/whl/cpu
+--extra-index-url https://pypi.org/simple
+torch
 gsplat
 
 # pycolmap for data parsing