Update requirements.txt

requirements.txt

@@ -7,9 +7,11 @@ huggingface_hub
 # Transformers for AI Model Loading
 transformers
 
-# PyTorch (Choose GPU or CPU version)
-torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118  # For GPU
-# torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu  # Uncomment for CPU-only
+# Torch (GPU version - CUDA 11.8)
+torch==2.2.0+cu118 torchvision==0.17.0+cu118 torchaudio==2.2.0+cu118 --index-url https://download.pytorch.org/whl/cu118
+
+# Uncomment the below line if running on CPU-only
+# torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
 
 # OpenAI's accelerate for faster model inference
 accelerate