Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 
-# Set environment variables to disable compilation and optimize PyTorch
 os.environ["TORCHDYNAMO_DISABLE"] = "1"
 os.environ["TORCH_COMPILE_DISABLE"] = "1"
 os.environ["PYTORCH_DISABLE_CUDNN_BENCHMARK"] = "1"
@@ -13,11 +12,11 @@ import spaces
 import logging
 from huggingface_hub import login
 
-# Disable PyTorch compilation completely
+
 torch._dynamo.config.disable = True
 torch._dynamo.config.suppress_errors = True
 
-# Set up logging
+
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
@@ -25,7 +24,7 @@ hf_token = os.getenv("HF_TOKEN")
 if hf_token:
     login(token=hf_token)
 
-# Global variables for model and speakers
+
 tts_model = None
 speakers_dict = None
 model_initialized = False
@@ -82,13 +81,13 @@ 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):
-    """Generate speech using the pre-loaded model"""
+
     
     if not text.strip():
         return None, "Please enter some Bambara text."
     
     try:
-        # Get the initialized model and speakers
+
         tts, speakers = initialize_model()
         
         speaker = speakers[speaker_name]
@@ -122,7 +121,7 @@ def generate_speech(text, speaker_name, use_advanced, temperature, top_k, top_p,
         logger.error(f"Speech generation failed: {e}")
         return None, f"❌ Error: {str(e)}"
 
-# Define speaker names for UI
+
 SPEAKER_NAMES = ["Adame", "Moussa", "Bourama", "Modibo", "Seydou"]
 
 examples = [
@@ -130,9 +129,9 @@ examples = [
     ["I ni ce", "Moussa"],
     ["Aw ni tile", "Bourama"],
     ["I ka kene wa?", "Modibo"],
-    ["Ala ka Mali suma", "Adame"],
+    ["Lakɔli karamɔgɔw tun tɛ ka se ka sɛbɛnni kɛ ka ɲɛ walanba kan wa denmisɛnw tun tɛ se ka o sɛbɛnni ninnu ye kuma tɛ ka kalan. Denmisɛnw kɛra kunfinw ye", "Adame"],
     ["sigikafɔ kɔnɔ jamanaw ni ɲɔgɔn cɛ, olu ye a haminankow ye, wa o ko ninnu ka kan ka kɛ sariya ani tilennenya kɔnɔ", "Seydou"],
-    ["Aw ni ce. Ne tɔgɔ ye Kaya Magan. Aw Sanbe Sanbe.", "Moussa"],
+    ["Aw ni ce. Ne tɔgɔ ye Adama. Awɔ ne ye Malien de ye. Aw Sanbɛ,  Sanbɛ. San min tɛ ɲinan ye, an bɛɛ ka jɛ ka o seli ɲɔgɔn fɛ,  hɛɛrɛ  ni lafiya la. Ala ka Mali suma. Ala ka Mali yiriwa. Ala ka Mali taa ɲɛ. Ala ka an ka seliw caya. Ala ka yafa an bɛɛ ma", "Moussa"],
     ["An dɔlakelen bɛ masike bilenman don ka tɔw gɛn.", "Bourama"],
     ["Aw ni ce. Seidu bɛ aw fo wa aw ka yafa a ma, ka da a kan tuma dɔw la kow ka can.", "Modibo"],
 ]
@@ -147,8 +146,7 @@ def build_interface():
         Convert Bambara text to speech using AI. This model is currently experimental.
         
         **Bambara** is spoken by millions of people in Mali and West Africa.
-        
-        ⚡ **Note**: Model loads automatically on first use and stays loaded for optimal performance.
+        .
         """)
         
         with gr.Row():
@@ -227,20 +225,20 @@ def build_interface():
             container=False
         )
         
-        with gr.Accordion("📚 Try These Examples", open=True):
+        with gr.Accordion( Try These Examples", open=True):
             def load_example(text, speaker):
                 return text, speaker, False, 0.8, 50, 0.9, 2048
             
             gr.Markdown("**Click any example below:**")
             
             for i, (text, speaker) in enumerate(examples):
-                btn = gr.Button(f"🎯 {text[:30]}{'...' if len(text) > 30 else ''}", size="sm")
+                btn = gr.Button(f" {text[:30]}{'...' if len(text) > 30 else ''}", size="sm")
                 btn.click(
                     fn=lambda t=text, s=speaker: load_example(t, s),
                     outputs=[text_input, speaker_dropdown, use_advanced, temperature, top_k, top_p, max_tokens]
                 )
         
-        with gr.Accordion("ℹ️ About", open=False):
+        with gr.Accordion(" About", open=False):
             gr.Markdown("""
             **⚠️ This is an experimental Bambara TTS model.**