load models only if duplicated and gou associated spaces

app.py

@@ -33,58 +33,61 @@ def trim_audio_to_5s_temp(audio_path, sample_rate=16000):
     sf.write(temp_path, audio, samplerate=sample_rate)
     return temp_path
 
-num_gpus = torch.cuda.device_count()
-print(f"GPU AVAILABLE: {num_gpus}")
-
-# Download All Required Models using `snapshot_download`
-
-# Download Wan2.1-I2V-14B-480P model
-wan_model_path = snapshot_download(
-    repo_id="Wan-AI/Wan2.1-I2V-14B-480P",
-    local_dir="./weights/Wan2.1-I2V-14B-480P",
-    #local_dir_use_symlinks=False
-)
-
-# Download Chinese wav2vec2 model
-wav2vec_path = snapshot_download(
-    repo_id="TencentGameMate/chinese-wav2vec2-base",
-    local_dir="./weights/chinese-wav2vec2-base",
-    #local_dir_use_symlinks=False
-)
-
-# Download MeiGen MultiTalk weights
-multitalk_path = snapshot_download(
-    repo_id="MeiGen-AI/MeiGen-MultiTalk",
-    local_dir="./weights/MeiGen-MultiTalk",
-    #local_dir_use_symlinks=False
-)
-
-# Define paths
-base_model_dir = "./weights/Wan2.1-I2V-14B-480P"
-multitalk_dir = "./weights/MeiGen-MultiTalk"
-
-# File to rename
-original_index = os.path.join(base_model_dir, "diffusion_pytorch_model.safetensors.index.json")
-backup_index = os.path.join(base_model_dir, "diffusion_pytorch_model.safetensors.index.json_old")
-
-# Rename the original index file
-if os.path.exists(original_index):
-    os.rename(original_index, backup_index)
-    print("Renamed original index file to .json_old")
-
-# Copy updated index file from MultiTalk
-shutil.copy2(
-    os.path.join(multitalk_dir, "diffusion_pytorch_model.safetensors.index.json"),
-    base_model_dir
-)
-
-# Copy MultiTalk model weights
-shutil.copy2(
-    os.path.join(multitalk_dir, "multitalk.safetensors"),
-    base_model_dir
-)
-
-print("Copied MultiTalk files into base model directory.")
+if is_gpu_associated:
+    num_gpus = torch.cuda.device_count()
+    print(f"GPU AVAILABLE: {num_gpus}")
+
+
+if not is_shared_ui and is_gpu_associated:
+    # Download All Required Models using `snapshot_download`
+
+    # Download Wan2.1-I2V-14B-480P model
+    wan_model_path = snapshot_download(
+        repo_id="Wan-AI/Wan2.1-I2V-14B-480P",
+        local_dir="./weights/Wan2.1-I2V-14B-480P",
+        #local_dir_use_symlinks=False
+    )
+
+    # Download Chinese wav2vec2 model
+    wav2vec_path = snapshot_download(
+        repo_id="TencentGameMate/chinese-wav2vec2-base",
+        local_dir="./weights/chinese-wav2vec2-base",
+        #local_dir_use_symlinks=False
+    )
+
+    # Download MeiGen MultiTalk weights
+    multitalk_path = snapshot_download(
+        repo_id="MeiGen-AI/MeiGen-MultiTalk",
+        local_dir="./weights/MeiGen-MultiTalk",
+        #local_dir_use_symlinks=False
+    )
+
+    # Define paths
+    base_model_dir = "./weights/Wan2.1-I2V-14B-480P"
+    multitalk_dir = "./weights/MeiGen-MultiTalk"
+
+    # File to rename
+    original_index = os.path.join(base_model_dir, "diffusion_pytorch_model.safetensors.index.json")
+    backup_index = os.path.join(base_model_dir, "diffusion_pytorch_model.safetensors.index.json_old")
+
+    # Rename the original index file
+    if os.path.exists(original_index):
+        os.rename(original_index, backup_index)
+        print("Renamed original index file to .json_old")
+
+    # Copy updated index file from MultiTalk
+    shutil.copy2(
+        os.path.join(multitalk_dir, "diffusion_pytorch_model.safetensors.index.json"),
+        base_model_dir
+    )
+
+    # Copy MultiTalk model weights
+    shutil.copy2(
+        os.path.join(multitalk_dir, "multitalk.safetensors"),
+        base_model_dir
+    )
+
+    print("Copied MultiTalk files into base model directory.")
 
 if not is_shared_ui: