Update app.py

app.py

@@ -11,6 +11,7 @@ import numpy as np
 import spaces
 import logging
 from huggingface_hub import login
+import threading
 
 
 torch._dynamo.config.disable = True
@@ -28,36 +29,38 @@ if hf_token:
 tts_model = None
 speakers_dict = None
 model_initialized = False
+model_initialized_lock = threading.Lock()
 
 @spaces.GPU()
 def initialize_model():
     """Initialize the TTS model and speakers - called once with GPU context"""
-    global tts_model, speakers_dict, model_initialized
-    
-    if not model_initialized:
-        logger.info("Initializing Bambara TTS model...")
-        
-        try:
-            from maliba_ai.tts.inference import BambaraTTSInference
-            from maliba_ai.config.speakers import Adame, Moussa, Bourama, Modibo, Seydou
-            
+    global tts_model, speakers_dict, model_initialized, model_initialized_lock
 
-            tts_model = BambaraTTSInference()
-            
-            speakers_dict = {
-                "Adame": Adame,
-                "Moussa": Moussa, 
-                "Bourama": Bourama,
-                "Modibo": Modibo,
-                "Seydou": Seydou
-            }
+    with model_initialized_lock :
+        if not model_initialized:
+            logger.info("Initializing Bambara TTS model...")
             
-            model_initialized = True
-            logger.info("Model initialized successfully!")
-            
-        except Exception as e:
-            logger.error(f"Failed to initialize model: {e}")
-            raise e
+            try:
+                from maliba_ai.tts.inference import BambaraTTSInference
+                from maliba_ai.config.speakers import Adame, Moussa, Bourama, Modibo, Seydou
+                
+    
+                tts_model = BambaraTTSInference()
+                
+                speakers_dict = {
+                    "Adame": Adame,
+                    "Moussa": Moussa, 
+                    "Bourama": Bourama,
+                    "Modibo": Modibo,
+                    "Seydou": Seydou
+                }
+                
+                model_initialized = True
+                logger.info("Model initialized successfully!")
+                
+            except Exception as e:
+                logger.error(f"Failed to initialize model: {e}")
+                raise e
     
     return tts_model, speakers_dict
 
@@ -78,14 +81,14 @@ def validate_inputs(text, temperature, top_k, top_p, max_tokens):
 
 @spaces.GPU()
 def generate_speech(text, speaker_name, use_advanced, temperature, top_k, top_p, max_tokens):
-
-    
+    global tts_model, speakers_dict, model_initialized, model_initialized_lock   
     if not text.strip():
         return None, "Please enter some Bambara text."
     
     try:
-
-        tts, speakers = initialize_model()
+        with model_initialized_lock :
+            if not model_initialized : 
+                tts, speakers = initialize_model()
         
         speaker = speakers[speaker_name]