Update Gradio version and optimize dependencies

requirements.txt

@@ -1,37 +1,36 @@
+# Core dependencies
+torch
+torchvision
+torchaudio
+transformers==4.33.1
+accelerate==0.22.0
+gradio>=4.0.0
+spaces
+
+# Model dependencies
 diffusers @ git+https://github.com/TMElyralab/diffusers.git@tme
 ip_adapter @ git+https://github.com/tencent-ailab/IP-Adapter.git@main
 clip @ git+https://github.com/openai/CLIP.git@main
 controlnet_aux @ git+https://github.com/TMElyralab/controlnet_aux.git@tme
 mmcm @ git+https://github.com/TMElyralab/MMCM.git@setup
-torch
-torchvision
-torchaudio
+
+# Additional dependencies
 ninja==1.11.1
-transformers==4.33.1
 bitsandbytes==0.41.1
-decord==0.6.0
-accelerate==0.22.0
 xformers==0.0.21
 omegaconf
 einops
 imageio==2.31.1
 pandas
-h5py
 matplotlib
-modelcards==0.1.6
 pynvml==11.5.0
 moviepy==1.0.3
 scikit-learn
-librosa
+librosa==0.10.1
 ffmpeg-python
 easydict
-webp
 mediapipe==0.10.3
-cython==3.0.2
 insightface==0.7.3
-librosa==0.10.1
-onnx==1.14.1
-onnxruntime==1.15.1
 opencv-python
 Pillow
 protobuf==3.20.3
@@ -45,14 +44,11 @@ pytorch-lightning==2.0.8
 timm==0.9.12
 safetensors==0.3.3
 pycocotools
-wget
 kornia==0.7.0
 open_clip_torch==2.20.0
 torchmetrics==1.1.1
 invisible-watermark==0.1.5
 ftfy==6.1.1
-gradio==3.50.2
 loguru==0.6.0
 markupsafe==2.0.1
-cuid
-spaces
+cuid

scripts/gradio/app_gradio_space.py

@@ -21,30 +21,28 @@ sys.path.insert(0, f"{ProjectDir}/diffusers/src")
 sys.path.insert(0, f"{ProjectDir}/controlnet_aux/src")
 sys.path.insert(0, f"{ProjectDir}/scripts/gradio")
 
-result = subprocess.run(
-    ["pip", "install", "--no-cache-dir", "-U", "openmim"],
-    capture_output=True,
-    text=True,
-)
-print(result)
-
-result = subprocess.run(["mim", "install", "mmengine"], capture_output=True, text=True)
-print(result)
-
-result = subprocess.run(
-    ["mim", "install", "mmcv>=2.0.1"], capture_output=True, text=True
-)
-print(result)
+# Install dependencies first
+def install_dependencies():
+    dependencies = [
+        "openmim",
+        "mmengine",
+        "mmcv>=2.0.1",
+        "mmdet>=3.1.0",
+        "mmpose>=1.1.0"
+    ]
+    for dep in dependencies:
+        try:
+            subprocess.run(
+                ["pip", "install", "--no-cache-dir", "-U", dep],
+                check=True,
+                capture_output=True,
+                text=True
+            )
+        except subprocess.CalledProcessError as e:
+            print(f"Warning: Failed to install {dep}: {e.stderr}")
 
-result = subprocess.run(
-    ["mim", "install", "mmdet>=3.1.0"], capture_output=True, text=True
-)
-print(result)
+install_dependencies()
 
-result = subprocess.run(
-    ["mim", "install", "mmpose>=1.1.0"], capture_output=True, text=True
-)
-print(result)
 ignore_video2video = True
 max_image_edge = 960