load_models

app.py

@@ -54,23 +54,30 @@ models_configs = {
 player = NemoAudioPlayer(Config())
 demo_examples = Demo()()
 models = {}
+for model_name, config in models_configs.items():
+    print(f"Loading {model_name}...")
+    models[model_name] = KaniModel(config, player, token_)
+    print(f"{model_name} loaded!")
+print("All models loaded!")
 
-def initialize_models():
-    """Initialize models globally to avoid reloading"""
-    global models
+
+
+# def initialize_models():
+#     """Initialize models globally to avoid reloading"""
+#     global models
     
-    # if player is None:
-    #     print("Initializing NeMo Audio Player...")
-    #     player = NemoAudioPlayer(Config())
-    #     print("NeMo Audio Player initialized!")
+#     # if player is None:
+#     #     print("Initializing NeMo Audio Player...")
+#     #     player = NemoAudioPlayer(Config())
+#     #     print("NeMo Audio Player initialized!")
     
-    if not models:
-        print("Loading TTS models...")
-        for model_name, config in models_configs.items():
-            print(f"Loading {model_name}...")
-            models[model_name] = KaniModel(config, player, token_)
-            print(f"{model_name} loaded!")
-        print("All models loaded!")
+#     if not models:
+#         print("Loading TTS models...")
+#         for model_name, config in models_configs.items():
+#             print(f"Loading {model_name}...")
+#             models[model_name] = KaniModel(config, player, token_)
+#             print(f"{model_name} loaded!")
+#         print("All models loaded!")
 
 @spaces.GPU
 def generate_speech_gpu(text, model_choice):
@@ -78,7 +85,7 @@ def generate_speech_gpu(text, model_choice):
     Generate speech from text using the selected model on GPU
     """
     # Initialize models if not already done
-    initialize_models()
+    # initialize_models()
     
     if not text.strip():
         return None, "Please enter text for speech generation."