Update Gradio/app.py

Gradio/app.py

@@ -11,21 +11,24 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "Urdu-ASR-flags")
 
-############## DagsHub ################################
+############## Inference ##############################
 
-Model = "kingabzpro/wav2vec2-large-xls-r-300m-Urdu"
-# This is not working because Huggingface has completely changed the git server. 
-# from dagshub.streaming import install_hooks
-# install_hooks()
 
-############## Inference ##############################
+transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-base.en")
+
+def transcribe(audio):
+    sr, y = audio
+    y = y.astype(np.float32)
+    y /= np.max(np.abs(y))
 
+    return transcriber({"sampling_rate": sr, "raw": y})["text"]
 
-def asr(audio):
 
-    asr = pipeline("automatic-speech-recognition", model=Model)
-    prediction = asr(audio, chunk_length_s=30)
-    return unicodedata.normalize("NFC",prediction["text"])
+demo = gr.Interface(
+    transcribe,
+    gr.Audio(sources=["microphone"]),
+    "text",
+)
 
 
 ################### Gradio Web APP ################################
@@ -47,19 +50,12 @@ article = "<p style='text-align: center'><a href='https://dagshub.com/kingabzpro
 examples = [["Sample/sample1.mp3"], ["Sample/sample2.mp3"], ["Sample/sample3.mp3"]]
 
 
-Input = gr.Audio(
-    source="microphone",
-    type="filepath",
-    label="Please Record Your Voice",
-)
-Output = gr.Textbox(label="Urdu Script")
-
 
 def main():
     iface = gr.Interface(
-        asr,
-        Input,
-        Output,
+        transcribe,
+        gr.Audio(sources=["microphone"]),
+        "text",
         title=title,
         allow_flagging="manual",
         flagging_callback=hf_writer,