Update requirements.txt

requirements.txt

@@ -1,10 +1,25 @@
-torch==2.1.0
-torchvision==0.16.0
-transformers==4.35.0
-diffusers==0.24.0
-accelerate==0.25.0
-scikit-learn==1.3.2
-gradio==4.14.0
-Pillow==10.1.0
-numpy==1.26.2
-xformers==0.0.23
+# 核心框架
+gradio>=4.0.0
+torch>=2.0.0
+torchvision>=0.15.0
+
+# 图像处理
+Pillow>=9.5.0
+opencv-python-headless>=4.8.0
+
+# AI模型相关
+transformers>=4.30.0
+diffusers>=0.20.0
+accelerate>=0.20.0
+safetensors>=0.3.0
+
+# 数据处理
+numpy>=1.24.0
+scikit-learn>=1.3.0
+scipy>=1.10.0
+
+# 可选优化库（如果GPU内存充足）
+# xformers>=0.0.20
+
+# 其他工具
+requests>=2.31.0