Update requirements.txt

requirements.txt

@@ -14,9 +14,7 @@ matplotlib==3.10.0           # Plotting and visualizations
 
 # === Machine Learning ===
 scikit-learn==1.6.1          # Evaluation metrics and preprocessing utilities
-torch==2.6.0          # PyTorch with CUDA 12.4 (GPU support)
-torchvision==0.21.0+cu124    # Optional (can support visual tasks, not critical here)
-torchaudio==2.6.0+cu124      # Optional (comes with torch install, can stay)
+torch==2.1.0
 
 # === Utilities ===
 tqdm==4.67.1                 # Progress bars