Update app.py

app.py

@@ -39,29 +39,39 @@ tts_voice_list = asyncio.get_event_loop().run_until_complete(edge_tts.list_voice
 tts_voices = [f"{v['ShortName']}-{v['Gender']}" for v in tts_voice_list]
 
 model_root = "weights"
-models = [
+
+# ---- Local models ----
+local_models = [
     d for d in os.listdir(model_root) if os.path.isdir(os.path.join(model_root, d))
 ]
-if len(models) == 0:
-    raise ValueError("No model found in `weights` folder")
-models.sort()
-
+if not local_models:
+    print("⚠️ No model found in local `weights` folder")
+local_models.sort()
 
-REPO_ID = "simpsonsaiorg/stream-models"  # your repo
+# ---- HF models ----
+REPO_ID = "simpsonsaiorg/stream-models"
 api = HfApi()
 
-# Get all files in the repo
 all_files = api.list_repo_files(REPO_ID)
-print('ALL FILES')
-print(all_files)
-# Group files by model folder
-#models = defaultdict(list)
+
+hf_models = defaultdict(list)
 for f in all_files:
     parts = f.split("/")
-    if len(parts) == 2:  # weights/model_name/file.pt
-        models[parts[0]].append(parts[1])
+    if len(parts) == 3 and parts[0] == "weights":
+        model_name, filename = parts[1], parts[2]
+        hf_models[model_name].append(filename)
+
+# ---- Merge / display ----
+available_models = sorted(set(local_models) | set(hf_models.keys()))
+
+print("Local models:", local_models)
+print("HF models:", list(hf_models.keys()))
+print("Available models (combined):", available_models)
+
+
 
 # Example: load a specific model (like your hubert loader)
+"""
 def load_model(model_name):
     if model_name not in models:
         raise ValueError(f"Model '{model_name}' not found in repo")
@@ -72,13 +82,9 @@ def load_model(model_name):
         loaded[file] = torch.load(path, map_location="cpu")
     return loaded
 
-# List available models
-available_models = list(models.keys())
-print("Available models:", available_models)
-
 # Load homer model
 #homer_model = load_model("homer")
-
+"""