Update app.py

app.py

@@ -38,14 +38,12 @@ def initialize_model():
         logger.info("Initializing Bambara TTS model...")
         
         try:
-            # Import inside GPU context to avoid CUDA initialization errors
             from maliba_ai.tts.inference import BambaraTTSInference
             from maliba_ai.config.speakers import Adame, Moussa, Bourama, Modibo, Seydou
             
-            # Initialize model
+
             tts_model = BambaraTTSInference()
             
-            # Initialize speakers
             speakers_dict = {
                 "Adame": Adame,
                 "Moussa": Moussa, 
@@ -64,7 +62,6 @@ def initialize_model():
     return tts_model, speakers_dict
 
 def validate_inputs(text, temperature, top_k, top_p, max_tokens):
-    """Validate user inputs"""
     if not text or not text.strip():
         return False, "Please enter some Bambara text."
     
@@ -126,28 +123,33 @@ SPEAKER_NAMES = ["Adame", "Moussa", "Bourama", "Modibo", "Seydou"]
 
 examples = [
     ["Aw ni ce", "Adame"],
-    ["I ni ce", "Moussa"],
-    ["Aw ni tile", "Bourama"],
-    ["I ka kene wa?", "Modibo"],
-    ["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"],
+    ["Mali bɛna diya kɔsɛbɛ,  ka a da a kan baara bɛ ka kɛ.", "Moussa"],
+    ["Ne bɛ se ka sɛbɛnni yɛlɛma ka kɛ kuma ye", "Bourama"],
+    ["I ka kɛnɛ wa?", "Modibo"],
+    ["Lakɔli karamɔgɔw tun tɛ ka se ka sɛbɛnni kɛ ka ɲɛ walanba kan wa denmisɛnw tun tɛ ka se ka o sɛbɛnni ninnu ye,  kuma tɛ ka u 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 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"],
+    ["Aw ni ce. Ne tɔgɔ ye Adama. Awɔ ne ye Maliden de ye. Aw Sanbe Sanbe. 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"],
+    
 ]
 
 def build_interface():
     """Build the Gradio interface for Bambara TTS"""
     
-    with gr.Blocks(title="Bambara TTS - EXPERIMENTAL", theme=gr.themes.Soft()) as demo:
+    with gr.Blocks(title="Bambara TTS - EXPERIMENTAL") as demo:
         gr.Markdown("""
         # 🎤 Bambara Text-to-Speech ⚠️ EXPERIMENTAL
         
-        Convert Bambara text to speech using AI. This model is currently experimental.
+        **Powered by MALIBA-AI**
+        
+        Convert Bambara text to speech. This model is currently experimental.
         
         **Bambara** is spoken by millions of people in Mali and West Africa.
         .
         """)
+
+
         
         with gr.Row():
             with gr.Column(scale=2):
@@ -155,8 +157,8 @@ def build_interface():
                     label="📝 Bambara Text",
                     placeholder="Type your Bambara text here...",
                     lines=3,
-                    max_lines=6,
-                    value="Aw ni ce"
+                    max_lines=10,
+                    value="I ni ce"
                 )
                 
                 speaker_dropdown = gr.Dropdown(
@@ -241,7 +243,6 @@ def build_interface():
         with gr.Accordion(" About", open=False):
             gr.Markdown("""
             **⚠️ This is an experimental Bambara TTS model.**
-            
             - **Languages**: Bambara (bm)  
             - **Speakers**: 5 different voice options
             - **Sample Rate**: 16kHz