Update pre_build.py

pre_build.py

@@ -47,6 +47,7 @@ def main():
     print("Starting pre-build setup for Hugging Face Spaces...")
 
     # Setup GroundingDINO
+    os.system("pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124")
     os.system("pip install -r requirements_hf.txt")
 
     if not setup_groundingdino():