Update app.py

app.py

@@ -37,10 +37,17 @@ def setup():
 def load_lang(langname):
     if langname=="Icelandic":
         df = corpora.ds_i
+        model_path="carlosdanielhernandezmena/wav2vec2-large-xlsr-53-icelandic-ep10-1000h"
     elif langname =="Faroese":
         df = corpora.ds_f
+        model_path = "carlosdanielhernandezmena/whisper-large-faroese-8k-steps-100h"
+
+    model_word_separator = '|'
+    model_blank_token = '[PAD]'
+    lang_aligner = ctcalign.CTCAligner(model_path,model_word_separator,model_blank_token)
+    
     df = df.drop(columns=['audio', 'speaker_id','duration'])
-    return df[:10]#(df, df[:50])
+    return (df[:10], lang_aligner) #(df, df[:50])
 
 
 def f1(langname):
@@ -69,12 +76,15 @@ bl = gr.Blocks()
 
 with bl:
 
-    lloadr = gr.Dropdown(["Faroese", "Icelandic"], label="Select a language")#, info="Loading the dataset takes some time")
+    lloadr = gr.Dropdown(["Faroese", "Icelandic"], label="Select a language",value="Icelandic")#, info="Loading the dataset takes some time")
 
+    alignr = gr.State(value=ctcalign.CTCAligner(model_path="../models/LVL/wav2vec2-large-xlsr-53-icelandic-ep10-1000h",model_word_separator = '|',model_blank_token = '[PAD]'))
+    
     with gr.Row():
         #invisidata = gr.DataFrame(interactive=False, visible=False)
         databrowser = gr.DataFrame(wrap=True, max_rows=50, interactive=False, overflow_row_behaviour='paginate')
 
+
     
     btn1 = gr.Button(value="The random prosody button")
     btn1.style(full_width=False, size="sm")
@@ -83,8 +93,10 @@ with bl:
 
     btn1.click(f1, [lloadr], pl1)
 
+
+
     
-    lloadr.change(load_lang,lloadr,[databrowser])
+    lloadr.change(load_lang,lloadr,[databrowser,alignr])
 
 
     gr.Markdown(