Upload app.py with huggingface_hub

app.py

@@ -1,6 +1,6 @@
 
 import gradio as gr
-import torch, yaml, os, sys, glob
+import torch, yaml, os, sys, glob, re
 import librosa
 import soundfile as sf
 import torchaudio
@@ -12,6 +12,37 @@ nltk.download("punkt_tab", quiet=True)
 
 # ── Clone StyleTTS2 ────────────────────────────────────────
 os.system("git clone https://github.com/yl4579/StyleTTS2 /app/StyleTTS2 2>/dev/null || true")
+
+# ── Patch models.py for PyTorch 2.6 weights_only ──────────
+models_path = "/app/StyleTTS2/models.py"
+with open(models_path) as f:
+    code = f.read()
+
+patched = re.sub(
+    r'torch\.load\(([^)]+)\)',
+    lambda m: m.group(0) if 'weights_only' in m.group(1)
+              else f'torch.load({m.group(1)}, weights_only=False)',
+    code
+)
+with open(models_path, "w") as f:
+    f.write(patched)
+
+# Also patch utils.py
+utils_path = "/app/StyleTTS2/utils.py"
+with open(utils_path) as f:
+    code2 = f.read()
+
+patched2 = re.sub(
+    r'torch\.load\(([^)]+)\)',
+    lambda m: m.group(0) if 'weights_only' in m.group(1)
+              else f'torch.load({m.group(1)}, weights_only=False)',
+    code2
+)
+with open(utils_path, "w") as f:
+    f.write(patched2)
+
+print("✅ Patched torch.load calls")
+
 sys.path.insert(0, "/app/StyleTTS2")
 os.chdir("/app/StyleTTS2")