Use Wav2Vec2-bert-augmented model

app.py

@@ -6,7 +6,7 @@ from transformers import pipeline
 from huggingface_hub import model_info
 #from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
 
-MODEL_NAME = "ihanif/wav2vec2-xls-r-300m-pashto"
+MODEL_NAME = "ihanif/w2v-bert2-pashto-augmented"
 lang = "ps"
 
 #load pre-trained model and tokenizer
@@ -87,18 +87,6 @@ mf_transcribe = gr.Interface(
     examples=examples,
 )
 
-yt_transcribe = gr.Interface(
-    fn=yt_transcribe,
-    inputs=[gr.inputs.Textbox(lines=1, placeholder="Paste the URL to a YouTube video here", label="YouTube URL")],
-    outputs=["html", "text"],
-    layout="horizontal",
-    theme="huggingface",
-    title="(Transcribe YouTube) د پښتو اتوماتیک وینا پیژندنه",
-    description=(
-        "مهرباني وکړئ د خپل غږ په کارولو سره د پښتو لیکلو لپاره لاندې اپلیکیشن وکاروئ. تاسو کولی شئ یو آډیو فایل اپلوډ کړئ یا په خپل وسیله مایکروفون وکاروئ. مهرباني وکړئ ډاډ ترلاسه کړئ چې تاسو اجازه ورکړې ده"
-    ),
-    allow_flagging="never",
-)
 
 with demo:
     gr.TabbedInterface([mf_transcribe, yt_transcribe], ["Transcribe Audio", "Transcribe YouTube"])