Update app.py

app.py

@@ -7,11 +7,13 @@ import re
 #import torchaudio
 
 # Initialize the speech recognition pipeline and transliterator
-p1 = pipeline(task="automatic-speech-recognition", model="cdactvm/w2v-bert-2.0-odia_v1")
+#p1 = pipeline(task="automatic-speech-recognition", model="cdactvm/w2v-bert-2.0-odia_v1")
 p2 = pipeline(task="automatic-speech-recognition", model="cdactvm/w2v-bert-2.0-hindi_v1")
-p3 = pipeline(task="automatic-speech-recognition", model="cdactvm/kannada_w2v-bert_model")
-p4 = pipeline(task="automatic-speech-recognition", model="cdactvm/telugu_w2v-bert_model")
-p5 = pipeline(task="automatic-speech-recognition", model="Sajjo/w2v-bert-2.0-bangala-gpu-CV16.0_v2")
+#p3 = pipeline(task="automatic-speech-recognition", model="cdactvm/kannada_w2v-bert_model")
+#p4 = pipeline(task="automatic-speech-recognition", model="cdactvm/telugu_w2v-bert_model")
+#p5 = pipeline(task="automatic-speech-recognition", model="Sajjo/w2v-bert-2.0-bangala-gpu-CV16.0_v2")
+p6 = pipeline(task="automatic-speech-recognition", model="cdactvm/hf-open-assames")
+p7 = pipeline(task="automatic-speech-recognition", model="cdactvm/w2v-assames")
 os.system('git clone https://github.com/irshadbhat/indic-trans.git')
 os.system('pip install ./indic-trans/.')
 
@@ -47,6 +49,18 @@ def transcribe_bangala(speech):
     if text is None:
         return "Error: ASR returned None"
     return text
+
+def transcribe_assamese_model1(speech):
+    text = p6(speech)["text"]
+    if text is None:
+        return "Error: ASR returned None"
+    return text
+    
+def transcribe_assamese_model2(speech):
+    text = p7(speech)["text"]
+    if text is None:
+        return "Error: ASR returned None"
+    return text
     
 def transcribe_odiya_eng(speech):
     trn = Transliterator(source='ori', target='eng', build_lookup=True)
@@ -153,6 +167,11 @@ def sel_lng(lng, mic=None, file=None):
        return transcribe_ban_eng(audio)
     elif lng == "Bangala":
        return transcribe_bangala(audio)
+    elif lng == "Assamese-Model1":
+       return transcribe_assamese_model1(audio)
+    elif lng == "Assamese-Model2":
+       return transcribe_assamese_model2(audio)
+        
     
 # Function to replace incorrectly spelled words
 def replace_words(sentence):
@@ -377,7 +396,8 @@ demo=gr.Interface(
       
     inputs=[
         
-        gr.Dropdown(["Hindi","Hindi-trans","Odiya","Odiya-trans","Kannada","Kannada-trans","Telugu","Telugu-trans","Bangala","Bangala-trans"],value="Hindi",label="Select Language"),
+        #gr.Dropdown(["Hindi","Hindi-trans","Odiya","Odiya-trans","Kannada","Kannada-trans","Telugu","Telugu-trans","Bangala","Bangala-trans"],value="Hindi",label="Select Language"),
+        gr.Dropdown(["Hindi","Hindi-trans","Assamese-Model1","Assamese-Model2"],value="Hindi",label="Select Language"),
         gr.Audio(sources=["microphone","upload"], type="filepath"),
         #gr.Audio(sources="upload", type="filepath"),
         #"state"