ihanif commited on
Commit
838b360
·
1 Parent(s): f0d0de7

add wav2vec model and Pashto instructions

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -5,7 +5,7 @@ import pytube as pt
5
  from transformers import pipeline
6
  from huggingface_hub import model_info
7
 
8
- MODEL_NAME = "ihanif/whisper-medium-ps-augmented" #this always needs to stay in line 8 :D sorry for the hackiness
9
  lang = "ps"
10
 
11
  device = 0 if torch.cuda.is_available() else "cpu"
@@ -67,11 +67,9 @@ mf_transcribe = gr.Interface(
67
  outputs="text",
68
  layout="horizontal",
69
  theme="huggingface",
70
- title="Whisper Demo: Transcribe Audio",
71
  description=(
72
- "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the the fine-tuned"
73
- f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
74
- " of arbitrary length."
75
  ),
76
  allow_flagging="never",
77
  )
@@ -82,11 +80,9 @@ yt_transcribe = gr.Interface(
82
  outputs=["html", "text"],
83
  layout="horizontal",
84
  theme="huggingface",
85
- title="Whisper Demo: Transcribe YouTube",
86
  description=(
87
- "Transcribe long-form YouTube videos with the click of a button! Demo uses the the fine-tuned checkpoint:"
88
- f" [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files of"
89
- " arbitrary length."
90
  ),
91
  allow_flagging="never",
92
  )
 
5
  from transformers import pipeline
6
  from huggingface_hub import model_info
7
 
8
+ MODEL_NAME = "ihanif/wav2vec2-xls-r-300m-pashto" #this always needs to stay in line 8 :D sorry for the hackiness
9
  lang = "ps"
10
 
11
  device = 0 if torch.cuda.is_available() else "cpu"
 
67
  outputs="text",
68
  layout="horizontal",
69
  theme="huggingface",
70
+ title="(Transcribe Audio) د پښتو اتوماتیک وینا پیژندنه",
71
  description=(
72
+ "مهرباني وکړئ د خپل غږ په کارولو سره د پښتو لیکلو لپاره لاندې اپلیکیشن وکاروئ. تاسو کولی شئ یو آډیو فایل اپلوډ کړئ یا په خپل وسیله مایکروفون وکاروئ. مهرباني وکړئ ډاډ ترلاسه کړئ چې تاسو اجازه ورکړې ده"
 
 
73
  ),
74
  allow_flagging="never",
75
  )
 
80
  outputs=["html", "text"],
81
  layout="horizontal",
82
  theme="huggingface",
83
+ title="(Transcribe YouTube) د پښتو اتوماتیک وینا پیژندنه",
84
  description=(
85
+ "مهرباني وکړئ د خپل غږ په کارولو سره د پښتو لیکلو لپاره لاندې اپلیکیشن وکاروئ. تاسو کولی شئ یو آډیو فایل اپلوډ کړئ یا په خپل وسیله مایکروفون وکاروئ. مهرباني وکړئ ډاډ ترلاسه کړئ چې تاسو اجازه ورکړې ده"
 
 
86
  ),
87
  allow_flagging="never",
88
  )