Update app.py

app.py

@@ -94,7 +94,7 @@ def get_speakers_dict():
 
 @spaces.GPU()
 def initialize_model_once():
-    """Initialize model with retry logic for Unsloth compilation issues"""
+    """Initialize model exactly like your old working version"""
     global _tts_model, _speakers_dict, _model_initialized, _initialization_in_progress
     
     if _model_initialized:
@@ -110,44 +110,34 @@ def initialize_model_once():
     
     _initialization_in_progress = True
     
-    # Retry logic for Unsloth compilation issues
-    max_retries = 2
-    for attempt in range(max_retries + 1):
-        try:
-            logger.info(f"Initializing Bambara TTS model... (attempt {attempt + 1})")
-            start_time = time.time()
-            
-            # Use the correct import path
-            from maliba_ai.tts.inference import BambaraTTSInference
-            
-            model = BambaraTTSInference()
-            speakers = get_speakers_dict()
-            
-            if not speakers:
-                raise ValueError("Failed to load speakers dictionary")
-            
-            _tts_model = model
-            _speakers_dict = speakers
-            _model_initialized = True
-            
-            elapsed = time.time() - start_time
-            logger.info(f"Model initialized successfully in {elapsed:.2f} seconds!")
-            
-            return _tts_model, _speakers_dict
-            
-        except Exception as e:
-            error_msg = str(e)
-            if "unsloth_compiled_module" in error_msg and attempt < max_retries:
-                logger.warning(f"Unsloth compilation failed (attempt {attempt + 1}/{max_retries + 1}), retrying...")
-                time.sleep(2)  # Brief delay before retry
-                continue
-            else:
-                logger.error(f"Failed to initialize model: {e}")
-                _initialization_in_progress = False
-                raise e
+    try:
+        logger.info("Initializing Bambara TTS model...")
+        start_time = time.time()
+        
+        # Use the correct import path
+        from maliba_ai.tts.inference import BambaraTTSInference
+        
+        model = BambaraTTSInference()
+        speakers = get_speakers_dict()
+        
+        if not speakers:
+            raise ValueError("Failed to load speakers dictionary")
+        
+        _tts_model = model
+        _speakers_dict = speakers
+        _model_initialized = True
+        
+        elapsed = time.time() - start_time
+        logger.info(f"Model initialized successfully in {elapsed:.2f} seconds!")
+        
+        return _tts_model, _speakers_dict
+        
+    except Exception as e:
+        logger.error(f"Failed to initialize model: {e}")
+        _initialization_in_progress = False
+        raise e
     finally:
-        if not _model_initialized:
-            _initialization_in_progress = False
+        _initialization_in_progress = False
 
 def validate_inputs(text, temperature, top_k, top_p, max_tokens):
     """Same validation as your old version"""