Update requirements.txt

requirements.txt

@@ -1,7 +1,4 @@
-# Ultralytics requirements
-# Usage: pip install -r requirements.txt
-
-# Base ----------------------------------------
+# Base requirements
 matplotlib>=3.2.2
 numpy>=1.21.6
 opencv-python>=4.6.0
@@ -9,46 +6,28 @@ Pillow>=7.1.2
 PyYAML>=5.3.1
 requests>=2.23.0
 scipy>=1.4.1
-torch>=1.7.0
-torchvision>=0.8.1
 tqdm>=4.64.0
-gradio==3.26.0
-ultralytics>=8.0.132
 urllib3>=1.26.14
-timm==0.6.12
-open3d==0.16.0
-open3d-python==0.3.0.0
 
-# Logging -------------------------------------
-# tensorboard>=2.4.1
-# clearml
-# comet
+# ZeroGPU specific requirements
+torch==2.4.0  # Specific version for ZeroGPU
+torchvision==0.15.0  # Compatible with torch 2.4.0
+gradio>=4.0.0  # Required for ZeroGPU
+spaces  # Required for ZeroGPU
 
-# Plotting ------------------------------------
+# Model specific
+ultralytics>=8.0.132
+timm==0.6.12
+
+# Visualization
+open3d==0.16.0
+plotly  # For 3D visualization
 pandas>=1.1.4
 seaborn>=0.11.0
 
-# Export --------------------------------------
-# coremltools>=6.0  # CoreML export
-# onnx>=1.12.0  # ONNX export
-# onnxsim>=0.4.1  # ONNX simplifier
-# nvidia-pyindex  # TensorRT export
-# nvidia-tensorrt  # TensorRT export
-# scikit-learn==0.19.2  # CoreML quantization
-# tensorflow>=2.4.1  # TF exports (-cpu, -aarch64, -macos)
-# tflite-support
-# tensorflowjs>=3.9.0  # TF.js export
-# openvino-dev>=2022.3  # OpenVINO export
-
-# Extras --------------------------------------
-psutil  # system utilization
-thop>=0.1.1  # FLOPs computation
-# ipython  # interactive notebook
-# albumentations>=1.0.3
-# pycocotools>=2.0.6  # COCO mAP
-# roboflow
+# Utilities
+psutil
+thop>=0.1.1
 
-# Zero GPUs --------------------------------------
-torch==2.4.0  # ZeroGPU compatible version
-gradio>=4.0.0
-spaces
+# Remove conflicting package
+# open3d-python==0.3.0.0  # Remove this as it conflicts with open3d