Update requirements.txt

requirements.txt

@@ -1,20 +1,20 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
 
-# PyTorch (CUDA 11.8 builds)
+# PyTorch (CUDA 11.8)
 torch==2.2.2+cu118
 torchvision==0.17.2+cu118
 torchaudio==2.2.2+cu118
 
-# Core inference stack
+# Core inference
 transformers==4.43.3
 accelerate==0.33.0
 safetensors==0.4.4
-bitsandbytes==0.43.1        # 4-bit/8-bit quant; stable on T4
+bitsandbytes==0.43.1
 
-# UI & client
+# UI & client (stable pair)
 gradio==4.44.0
-gradio_client==1.3.0
-websockets==11.0.3             # fix for ModuleNotFoundError: websockets.asyncio
+gradio_client==0.17.0
+websockets>=11.0,<12.0  # stays on 11.x which works with gradio_client 0.17.x
 
 # Data / utils
 pandas==2.2.2
@@ -22,5 +22,6 @@ numpy==1.26.4
 regex==2024.7.24
 huggingface_hub==0.24.6
 
-# Tokenization / sentencepiece models support
+# Tokenization
 sentencepiece==0.1.99
+