Update

requirements.txt

@@ -9,15 +9,16 @@ pyyaml>=6.0.2
 gradio>=5.49.1
 huggingface_hub>=0.24.0
 spaces>=0.28.0
-torchmetrics>=1.8.2
 
-# Note: torch and torchvision are pre-installed by HuggingFace Spaces
-# DO NOT specify them here or you'll overwrite the correct version
+# PyTorch with CUDA 12.4 support (install first for compatibility)
+--extra-index-url https://download.pytorch.org/whl/cu124
+torch==2.6.0
+torchvision==0.21.0
+torchmetrics>=1.8.2
 
-# Pre-built wheels compatible with HuggingFace Spaces PyTorch
-# Spaces pre-installs torch==2.8.0 (PyPI version) - using 2.7.0 wheel (closest available, should be ABI compatible)
-# Python 3.10 (cp310), Linux x86_64, CUDA 12.8
-fused_ssim @ https://miropsota.github.io/torch_packages_builder/fused-ssim/fused_ssim-0.0.1%2Bb4fd832pt2.7.0cu128-cp310-cp310-linux_x86_64.whl
-gsplat @ https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/gsplat-0.1.0+torch2.6-cp310-cp310-linux_x86_64.whl
+# Pre-built wheels from HuggingFace (PyTorch 2.6.0 + CUDA 12.4)
+# Built for Python 3.10 (cp310) - compatible with HuggingFace Spaces Gradio SDK
+fused_ssim @ https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/fused_ssim-0.0.1%2Bb4fd832pt2.6.0cu124-cp310-cp310-linux_x86_64.whl
+gsplat @ https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl