Update app.py

app.py

@@ -7,21 +7,20 @@ from TTS.utils.synthesizer import Synthesizer
 # Max input text length
 MAX_TXT_LEN = 400
 
-# Checkpoints from your repo
+# Available checkpoint(s) from your repo
 MODEL_INFO = {
-    # "checkpoint_2080000": "checkpoint_2080000.pth",
-    # "checkpoint_2085000": "checkpoint_2085000.pth",
-    # "checkpoint_2090000": "checkpoint_2090000.pth",
-    # "checkpoint_2095000": "checkpoint_2095000.pth",
     "checkpoint_2100000": "checkpoint_2100000.pth",
+    # If you want to add more later, just extend this dict
+    # "checkpoint_2095000": "checkpoint_2095000.pth",
+    # "checkpoint_2090000": "checkpoint_2090000.pth",
 }
 
 # Download config once
-config_path = hf_hub_download("huggingface.co/sulaimank/luganda_LMs/", filename="config.json")
+config_path = hf_hub_download("sulaimank/luganda_LMs", filename="config.json")
 
 # Function to load synthesizer on demand
 def load_synth(model_file):
-    model_path = hf_hub_download("huggingface.co/sulaimank/luganda_LMs/", filename=model_file)
+    model_path = hf_hub_download("sulaimank/luganda_LMs", filename=model_file)
     return Synthesizer(
         tts_checkpoint=model_path,
         tts_config_path=config_path
@@ -40,13 +39,29 @@ def tts(text: str, model_choice: str):
         synthesizer.save_wav(wav, fp.name)
         return fp.name
 
-# Example sentences
+# Example sentences (must match available model choices)
 examples = [
-    ["Oli otya, nsanyuse okukulaba.", "checkpoint_2080000"],
-    ["Abantu abangi mu Uganda bayogera Oluganda.", "checkpoint_2095000"],
+    ["Oli otya, nsanyuse okukulaba.", "checkpoint_2100000"],
+    ["Abantu abangi mu Uganda bayogera Oluganda.", "checkpoint_2100000"],
 ]
 
 description = """
 🗣️ **Luganda TTS Demo** 🗣️  
 Convert Luganda text into speech using VITS models trained with Coqui TTS.  
 """
+
+iface = gr.Interface(
+    fn=tts,
+    inputs=[
+        gr.Textbox(label="Enter Luganda Text", value="Gyebale ko nnyo."),
+        gr.Radio(label="Choose Model", choices=list(MODEL_INFO.keys()), value="checkpoint_2100000"),
+    ],
+    outputs=gr.Audio(label="Generated Speech", type="filepath"),
+    examples=examples,
+    title="Luganda TTS",
+    description=description,
+    live=False,
+)
+
+if __name__ == "__main__":
+    iface.launch()