Update app.py

app.py

@@ -13,6 +13,7 @@ import re
 p1 = pipeline(task="automatic-speech-recognition", model="cdactvm/w2v-bert-2.0-odia_v1")
 odia_model2 = pipeline(task="automatic-speech-recognition", model="cdactvm/w2v-bert-odia_v1")
 p2 = pipeline(task="automatic-speech-recognition", model="cdactvm/w2v-bert-2.0-hindi_v1")
+punjaib_modle_30000=pipeline(task="automatic-speech-recognition", model="cdactvm/wav2vec-bert-punjabi-30000")
 #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")
@@ -264,6 +265,25 @@ def process_transcription(input_sentence):
 
 ###########################################
 
+def transcribe_punjabi_30000(speech):
+    text = punjaib_modle_30000(speech)["text"]
+    if text is None:
+        return "Error: ASR returned None"
+    return text
+
+def transcribe_punjabi_eng_model_30000(speech):
+    trn = Transliterator(source='pun', target='eng', build_lookup=True)
+    text = punjaib_modle_30000(speech)["text"]
+    if text is None:
+        return "Error: ASR returned None"
+    sentence = trn.transform(text)
+    if sentence is None:
+        return "Error: Transliteration returned None"
+    replaced_words = replace_words(sentence)
+    processed_sentence = process_doubles(replaced_words)
+    return process_transcription(processed_sentence)
+
+###########################################
 def transcribe_odiya(speech):
     text = p1(speech)["text"]
     if text is None:
@@ -427,6 +447,10 @@ def sel_lng(lng, mic=None, file=None):
        return transcribe_odiya_model2(audio)
     elif lng == "Odia_trans_model2":
        return transcribe_odiya_eng_model2(audio)
+    elif lng == "punjabi":
+       return transcribe_punjabi_30000(audio)
+    elif lng == "punjabi_trans":
+       return transcribe_punjabi_eng_model_30000(audio)
         
     
 
@@ -460,7 +484,7 @@ 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","Odia_model2","Odia_trans_model2","Assamese-LM","Assamese-Model2"],value="Hindi",label="Select Language"),
+        gr.Dropdown(["Hindi","Hindi-trans","Odiya","Odiya-trans","Odia_model2","Odia_trans_model2","Assamese-LM","Assamese-Model2","punjabi","punjabi-trans"],value="Hindi",label="Select Language"),
         gr.Audio(sources=["microphone","upload"], type="filepath"),
         #gr.Audio(sources="upload", type="filepath"),
         #"state"