Update requirements.txt

requirements.txt

@@ -1,27 +1,37 @@
-# PyTorch and related libraries
-torch>=1.10.0
-transformers>=4.21.0
-diffusers>=0.10.0
-torchvision==0.14.1
+# Ensure pip, setuptools, wheel are up-to-date before installing
+# pip install --upgrade pip setuptools wheel
 
-# Gradio for the web UI
-gradio>=3.16.2
+# Pin numpy below 2.x to avoid the "compiled with NumPy 1.x" error
+numpy<2.0
+
+# Some libraries may need a newer pybind11 to rebuild if needed
+pybind11>=2.12
+
+# PyTorch 2.1.0 includes torch.distributed.tensor
+# Choose +cpu or +cu118 variants if needed for GPU support.
+torch==2.1.0
+torchvision==0.16.0
+
+# Match Transformers with the above. 4.30+ uses torch.distributed.tensor, so it works with torch>=2.1
+transformers==4.31.0
 
-# Object detection
+# Object Detection
 ultralytics>=8.0.0
 
+# RealESRGAN dependencies
+realesrgan==0.3.0
+basicsr==1.4.2
+
+# Gradio for the web UI
+gradio>=3.16.2
+
 # Image processing
 opencv-python>=4.5.3.56
 Pillow>=9.0.0
-numpy>=1.21.0
 
-# Wikipedia lookup
+# Wikipedia
 wikipedia>=1.4.0
 
-# Upscaling model dependencies
-realesrgan>=0.3.0
-basicsr>=1.3.2
-
 # Additional utilities
 requests>=2.26.0
 huggingface_hub>=0.10.1