Update app.py

app.py

@@ -10,10 +10,9 @@ import torch
 import librosa #For converting audio sample rate to 16k
 from easymms.models.tts import TTSModel #For TTS inference using EasyMMS
 
-LANG = "dtp"
+LANG = "dtp" #Change to tih for Timugon Murut or iba for Iban
 model_id = "facebook/mms-1b-all"
 
-#Set target language to dtp (Kadazandusun)
 processor = AutoProcessor.from_pretrained(model_id)
 model = Wav2Vec2ForCTC.from_pretrained(model_id).to("cpu")
 processor.tokenizer.set_target_lang(LANG)
@@ -44,23 +43,15 @@ def transcribe(input): #Gradio UI wrapper function
 with gr.Blocks(theme = gr.themes.Soft()) as demo:
     gr.HTML(
         """
-<h1 align="center">Ponutun Tuturan om Pomorolou Sinuat Boros Dusun</h1>
-<h5 align="center">  Poomitanan kopogunaan do somit tutun tuturan om pomorolou sinuat (speech recognition and text-to-speech models)
-  pinoluda' di Woyotanud Tuturan Gumukabang Tagayo di Meta (Meta Massive Multilingual Speech Project)</h5>
-<h6 align = "center">Guguno (app) diti winonsoi di Ander © 2023 id Universiti Teknologi PETRONAS</h6>
-<style>
-    .container {
-      display: grid;
-      grid-template-columns:20% 5% 20%;
-      align-items: center;
-    }
-</style>
-<h6 align = "center">
-<div class = "container">
-    <div class = "image"> <a href='https://github.com/andergisomon/dtp-nlp-demo'><img src='https://img.shields.io/badge/Github-Code-success'></a> </div>
-    <div class = "image"></div>
-    <div class = "image"> <a href='https://huggingface.co/spaces/anderbogia/dtp-asr-demo-v2/'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue'></a> </div>
-</div></h6>
+            <h1 align="center">Ponutun Tuturan om Pomorolou Sinuat Boros Dusun</h1>
+            <h5 align="center">  Poomitanan kopogunaan do somit tutun tuturan om pomorolou sinuat (speech recognition and text-to-speech models)
+              pinoluda' di Woyotanud Tuturan Gumukabang Tagayo di Meta (Meta Massive Multilingual Speech Project)</h5>
+            <h6 align = "center">Guguno (app) diti winonsoi di Ander © 2023 id Universiti Teknologi PETRONAS</h6>
+
+            <div style='display:flex; gap: 0.25rem; '>
+                <div class = "image"> <a href='https://github.com/andergisomon/dtp-nlp-demo'><img src='https://img.shields.io/badge/Github-Code-success'></a> </div>
+                <div class = "image"> <a href='https://huggingface.co/spaces/anderbogia/dtp-asr-demo-v2/'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue'></a> </div>
+            </div>
         """)
 
     tts = TTSModel(LANG)
@@ -73,7 +64,7 @@ with gr.Blocks(theme = gr.themes.Soft()) as demo:
 
     with gr.Row():
       with gr.Column(scale = 1):
-          gr.HTML("""<h1 align="center"><img src="https://andergisomon.github.io/dtp-nlp-demo/huminodun_dall_e.png", alt="Video-LLaMA" border="0" style="margin: 0 auto; height: 200px;" /></a></h1>""")
+          gr.HTML("""<h1 align="center"><img src="https://andergisomon.github.io/dtp-nlp-demo/huminodun_dall_e.png", alt="" border="0" style="margin: 0 auto; height: 200px;" /></a></h1>""")
 
           gr.Markdown("""
           **Huminodun, nulai di somit pongulai kikito DALL-E**