Use install bash script instead of python subprocesses

app.py

@@ -30,18 +30,7 @@ from moviepy.editor import *
 from transformers import WhisperModel
 
 if os.getenv('SYSTEM') == 'spaces':
-    os.environ['HOME'] = '/home/user/app'
-
-    subprocess.call('pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118'.split())
-
-    subprocess.call('mim install mmengine'.split())
-    subprocess.call('mim install mmcv==2.0.1'.split())
-    subprocess.call('mim install mmdet==3.1.0'.split())
-    subprocess.call('mim install mmpose==1.1.0'.split())
-
-    subprocess.call('ls ~/.cache/huggingface/hub/'.split())
-    subprocess.call('ls ~/.cache/huggingface/hub/TMElyralab'.split())
-    subprocess.call('ls ~/.cache/huggingface/hub/TMElyralab/MuseTalk'.split())
+    subprocess.run(["bash", "install.sh"])
 
 ProjectDir = os.path.abspath(os.path.dirname(__file__))
 CheckpointsDir = os.path.join(ProjectDir, "models")

install.sh

@@ -1,3 +1,31 @@
 #!/bin/bash
 
-#./download_weights.sh
+# Since we use preloaded, cached versions of models, we must link them for further use
+mkdir ~/app/models/musetalkV15
+ln -s ~/.cache/huggingface/hub/models--TMElyralab--MuseTalk/snapshots/*/musetalkV15/* ~/app/models/musetalkV15/
+
+mkdir ~/app/models/musetalk
+ln -s ~/.cache/huggingface/hub/models--TMElyralab--MuseTalk/snapshots/*/musetalk/* ~/app/models/musetalk/
+
+mkdir ~/app/models/dwpose
+ln -s ~/.cache/huggingface/hub/models--yzd-v--DWPose/snapshots/*/* ~/app/models/dwpose/
+
+mkdir ~/app/models/sd-vae
+ln -s ~/.cache/huggingface/hub/models--stabilityai--sd-vae-ft-mse/snapshots/*/* ~/app/models/sd-vae/
+
+mkdir ~/app/models/syncnet
+ln -s ~/.cache/huggingface/hub/models--ByteDance--LatentSync/snapshots/*/* ~/app/models/syncnet/
+
+mkdir ~/app/models/whisper
+ln -s ~/.cache/huggingface/hub/models--openai--whisper-tiny/snapshots/*/* ~/app/models/whisper/
+
+# Hugging face only allows to install torch atleast version 2.1.0, so we reinstalling it at runtime
+pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
+
+# Mim uses ~/.cache/mim by default for cache, but hugging face only allows to use /home/user/app, so we use little hack
+export HOME=/home/user/app
+
+mim install mmengine
+mim install mmcv==2.0.1
+mim install mmdet==3.1.0
+mim install mmpose==1.1.0