Update requirements.txt

requirements.txt

@@ -1,12 +1,32 @@
-torch
-torchvision
-torchaudio
-diffusers==0.26.0
-transformers==4.38.0
-accelerate
-gradio
-lpips
-git+https://github.com/openai/CLIP.git
-bert-score
-Pillow
-numpy
+
+
+# Core ML libraries
+torch>=2.1.0
+torchvision>=0.16.0
+
+# Diffusers & Transformers
+diffusers>=0.23.1
+transformers>=4.35.0
+sentencepiece>=0.1.99
+
+# Hugging Face pipelines
+torchmetrics>=1.1.0
+accelerate>=0.24.0
+
+# Image quality & metrics
+lpips>=0.1.4
+clip @ git+https://github.com/openai/CLIP.git
+
+# NLP evaluation
+bert-score>=0.3.12
+
+# Image processing
+Pillow>=10.0.0
+torchvision>=0.16.0
+
+# Gradio
+gradio>=3.50.0
+
+# Optional GPU dependencies
+scipy>=1.11.0
+numpy>=1.26.0