Update app.py

app.py

@@ -13,7 +13,7 @@ import numpy as np
 import librosa
 import re_matching
 from tools.sentence import split_by_language
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, list_repo_files
 
 from clap_wrapper import get_clap_audio_feature, get_clap_text_feature
 from models import SynthesizerTrn
@@ -121,8 +121,27 @@ if __name__ == "__main__":
         if not info['enable']:
             continue
         name, title, repid, example, filename = info['name'], info['title'], info['repid'], info['example'], info['filename']
-        config_path = hf_hub_download(repo_id=repid, filename="config.json")
-        model_path = hf_hub_download(repo_id=repid, filename=f"{filename}")
+
+        # 1. ลิสต์ไฟล์ใน repo ทั้งหมด
+        files = list_repo_files(repo_id=repid)
+        
+        # 2. หา subfolder ที่มี model file อยู่
+        model_subfolder = None
+        for f in files:
+            if f.endswith(filename):
+                # แยก path ออกมาเพื่อดูว่าอยู่ใน subfolder ไหม
+                parts = f.split("/")
+                if len(parts) > 1:
+                    model_subfolder = "/".join(parts[:-1])
+                break  # เอาอันแรกที่เจอก็พอ
+        
+        # 3. โหลดไฟล์ตาม path ที่ได้
+        if model_subfolder:
+            model_path = hf_hub_download(repo_id=repid, filename=filename, subfolder=model_subfolder)
+            config_path = hf_hub_download(repo_id=repid, filename="config.json", subfolder=model_subfolder)
+        else:
+            model_path = hf_hub_download(repo_id=repid, filename=filename)
+            config_path = hf_hub_download(repo_id=repid, filename="config.json")
         hps = utils.get_hparams_from_file(config_path)
         version = hps.version if hasattr(hps, "version") else "v2"
         net_g = get_net_g(model_path, version, device, hps)