Update requirements.txt

requirements.txt

@@ -10,8 +10,9 @@ tqdm>=4.64.0
 urllib3>=1.26.14
 
 # ZeroGPU specific requirements
-torch==2.4.0  # Specific version for ZeroGPU
-torchvision==0.15.0  # Compatible with torch 2.4.0
+--extra-index-url https://download.pytorch.org/whl/cu118
+torch==2.0.1+cu118  # ZeroGPU compatible version
+torchvision==0.15.2+cu118  # Compatible with torch 2.0.1
 gradio>=4.0.0  # Required for ZeroGPU
 spaces  # Required for ZeroGPU
 
@@ -27,7 +28,4 @@ seaborn>=0.11.0
 
 # Utilities
 psutil
-thop>=0.1.1
-
-# Remove conflicting package
-# open3d-python==0.3.0.0  # Remove this as it conflicts with open3d
+thop>=0.1.1