Update app.py

app.py

@@ -3,18 +3,14 @@ from transformers import VitsModel, AutoTokenizer
 import torch
 import logging
 
-
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-
 languages = ["bambara", "boomu", "dogon", "pular", "songhoy", "tamasheq"]
 
-
 models = {}
 tokenizers = {}
 
-
 examples = {
     "bambara": "An filɛ ni ye yɔrɔ minna ni an ye an sigi ka a layɛ yala an bɛ ka baara min kɛ ɛsike a kɛlen don ka Ɲɛ wa ?",
     "boomu": "Vunurobe wozomɛ pɛɛ, Poli we zo woro han Deeɓenu wara li Deeɓenu faralo zuun. Lo we baba a lo wara yi see ɓa Zuwifera ma ɓa Gɛrɛkela wa.",
@@ -24,7 +20,6 @@ examples = {
     "tamasheq": "Toḍă tăfukt ɣas, issăɣră-dd măssi-s n-ašĕkrĕš ănaẓraf-net, inn'-as: 'Ǝɣĕr-dd inaxdimăn, tĕẓlĕd-asăn, sănt s-wi dd-ĕšrăynen har tĕkkĕd wi dd-ăzzarnen."
 }
 
-
 try:
     for lang in languages:
         logger.info(f"Loading model and tokenizer for {lang}...")
@@ -40,18 +35,14 @@ def generate_audio(language, text):
         return None, "Please enter some text to synthesize."
     
     try:
-  
         model = models[language]
         tokenizer = tokenizers[language]
         
-
         inputs = tokenizer(text, return_tensors="pt")
         
-
         with torch.no_grad():
             output = model(**inputs).waveform
         
-
         waveform = output.squeeze().cpu().numpy()
         sample_rate = model.config.sampling_rate
         
@@ -63,28 +54,26 @@ def generate_audio(language, text):
 def load_example(language):
     return examples.get(language, "No example available")
 
+with gr.Blocks(title="MalianVoices") as demo:
+    gr.Markdown(
+        """
+        # MalianVoices: 🇲🇱 Text-to-Speech in Six Malian Languages
 
-with gr.Blocks(title="Malian Languages TTS") as demo:
-    with gr.Blocks(title="MalianVoices") as demo:
-        gr.Markdown(
-                """
-                # MalianVoices: 🇲🇱 Text-to-Speech in Six Malian Languages
-        
-                Lightweight TTS for six Malian languages: **Bambara, Boomu, Dogon, Pular, Songhoy, Tamasheq**.
-        
-                - ✅ Real-time TTS with fast response
-                - ✅ Runs on CPU
-        
-                ## How to Use
-                1. Pick a language from the dropdown
-                2. Enter your text or load an example
-                3. Click **"Generate Audio"** to listen
-        
-                By [sudoping01](https://huggingface.co/sudoping01), from [sudoping01/malian-tts](https://huggingface.co/sudoping01/malian-tts). Fine-tuned on Meta’s MMS, CC BY-NC 4.0, non-commercial.
-                No setup. Type and hear it.
-                """
-         )
-        
+        Lightweight TTS for six Malian languages: **Bambara, Boomu, Dogon, Pular, Songhoy, Tamasheq**.
+
+        - ✅ Real-time TTS with fast response
+        - ✅ Runs on CPU
+
+        ## How to Use
+        1. Pick a language from the dropdown
+        2. Enter your text or load an example
+        3. Click **"Generate Audio"** to listen
+
+        By [sudoping01](https://huggingface.co/sudoping01), from [sudoping01/malian-tts](https://huggingface.co/sudoping01/malian-tts). Fine-tuned on Meta’s MMS, CC BY-NC 4.0, non-commercial.
+        No setup. Type and hear it.
+        """
+    )
+    
     with gr.Row():
         language = gr.Dropdown(choices=languages, label="Language", value="bambara")
         with gr.Column():
@@ -107,5 +96,4 @@ with gr.Blocks(title="Malian Languages TTS") as demo:
         outputs=text
     )
 
-
 demo.launch()