Update app.py

app.py

@@ -11,7 +11,7 @@ import numpy as np
 import spaces
 import logging
 from huggingface_hub import login
-
+import threading
 
 torch._dynamo.config.disable = True
 torch._dynamo.config.suppress_errors = True
@@ -28,36 +28,39 @@ 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
+    global tts_model, speakers_dict, model_initialized, model_initialized_lock
     
-    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
-            
-
-            tts_model = BambaraTTSInference()
-            
-            speakers_dict = {
-                "Adame": Adame,
-                "Moussa": Moussa, 
-                "Bourama": Bourama,
-                "Modibo": Modibo,
-                "Seydou": Seydou
-            }
+    # Always acquire lock first for async safety
+    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
+                
+                # All initialization inside the lock
+                tts_model = BambaraTTSInference()
+                speakers_dict = {
+                    "Adame": Adame,
+                    "Moussa": Moussa, 
+                    "Bourama": Bourama,
+                    "Modibo": Modibo,
+                    "Seydou": Seydou
+                }
+                
+                # Set flag last, after everything is ready
+                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