Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 import subprocess,os
 from datasets import load_dataset, Audio
-import corpora
-import ctcalign,graph
+import datas,ctcalign,graph
 from numpy import random
 
 
@@ -37,26 +36,22 @@ setup()
 
 def load_lang(langname):
     if langname=="Icelandic":
-        df = corpora.ds_i
-        model_path="carlosdanielhernandezmena/wav2vec2-large-xlsr-53-icelandic-ep10-1000h"
+        df = datas.ds_i
+        lang_aligner = datas.a_i
     elif langname =="Faroese":
-        df = corpora.ds_f
-        model_path = "carlosdanielhernandezmena/wav2vec2-large-xlsr-53-faroese-100h"
-
-    model_word_separator = '|'
-    model_blank_token = '[PAD]'
-    lang_aligner = ctcalign.aligner(model_path,model_word_separator,model_blank_token)
+        df = datas.ds_f
+        lang_aligner = datas.a_f
 
     df = df.data.to_pandas()
     df = df.drop(columns=['audio', 'speaker_id','duration'])
-    return (df[:10], lang_aligner) #(df, df[:50])
+    return (df[:15], lang_aligner) #(df, df[:50])
 
 
 def f1(langname,lang_aligner):
     if langname=="Icelandic":
-        ds = corpora.ds_i
+        ds = datas.ds_i
     elif langname =="Faroese":
-        ds = corpora.ds_f
+        ds = datas.ds_f
 
     
     #fig = plt.figure(figsize=(10,4))