razhan commited on
Commit
4987242
·
verified ·
1 Parent(s): 0fcdb29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -176,7 +176,7 @@ from transformers.pipelines.audio_utils import ffmpeg_read
176
  import tempfile
177
  import os
178
 
179
- MODEL_NAME = "razhan/whisper-base-hawrami"
180
  BATCH_SIZE = 1
181
  FILE_LIMIT_MB = 10
182
  YT_LENGTH_LIMIT_S = 60 * 10 # limit to 1 hour YouTube files
@@ -192,7 +192,7 @@ pipe = pipeline(
192
 
193
 
194
  # @spaces.GPU
195
- def transcribe(inputs, task="translate"):
196
  if inputs is None:
197
  raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
198
 
@@ -261,7 +261,7 @@ def _return_yt_html_embed(yt_url):
261
  # return html_embed_str, text
262
 
263
 
264
- def yt_transcribe(yt_url, task="translate", progress=gr.Progress(), max_filesize=75.0):
265
  progress(0, desc="Loading audio file...")
266
  html_embed_str = _return_yt_html_embed(yt_url)
267
  try:
@@ -298,7 +298,7 @@ mf_transcribe = gr.Interface(
298
  # gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
299
  ],
300
  outputs="text",
301
- title="Whisper Central Kurdish‌ (Sorani) Demo: Transcribe Audio",
302
  description=(
303
  "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the"
304
  f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
@@ -314,7 +314,7 @@ file_transcribe = gr.Interface(
314
  # gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
315
  ],
316
  outputs="text",
317
- title="Whisper Hawrami Demo: Translate Audio",
318
  description=(
319
  "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the"
320
  f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
@@ -330,7 +330,7 @@ yt_transcribe = gr.Interface(
330
  # gr.Radio(["transcribe", "translate"], label="Task", value="transcribe")
331
  ],
332
  outputs=["html", "text"],
333
- title="Whisper Hawrami Demo: Translate YouTube",
334
  description=(
335
  "Transcribe long-form YouTube videos with the click of a button! Demo uses the checkpoint"
336
  f" [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe video files of"
 
176
  import tempfile
177
  import os
178
 
179
+ MODEL_NAME = "razhan/whisper-base-hawrami-transcription"
180
  BATCH_SIZE = 1
181
  FILE_LIMIT_MB = 10
182
  YT_LENGTH_LIMIT_S = 60 * 10 # limit to 1 hour YouTube files
 
192
 
193
 
194
  # @spaces.GPU
195
+ def transcribe(inputs, task="transcribe"):
196
  if inputs is None:
197
  raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
198
 
 
261
  # return html_embed_str, text
262
 
263
 
264
+ def yt_transcribe(yt_url, task="transcribe", progress=gr.Progress(), max_filesize=75.0):
265
  progress(0, desc="Loading audio file...")
266
  html_embed_str = _return_yt_html_embed(yt_url)
267
  try:
 
298
  # gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
299
  ],
300
  outputs="text",
301
+ title="Whisper Horami Demo: Transcribe Audio",
302
  description=(
303
  "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the"
304
  f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
 
314
  # gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),
315
  ],
316
  outputs="text",
317
+ title="Whisper Horami Demo: Transcribe Audio",
318
  description=(
319
  "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the"
320
  f" checkpoint [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
 
330
  # gr.Radio(["transcribe", "translate"], label="Task", value="transcribe")
331
  ],
332
  outputs=["html", "text"],
333
+ title="Whisper Horami Demo: Translate YouTube",
334
  description=(
335
  "Transcribe long-form YouTube videos with the click of a button! Demo uses the checkpoint"
336
  f" [{MODEL_NAME}](https://huggingface.co/{MODEL_NAME}) and 🤗 Transformers to transcribe video files of"