CsanadT commited on
Commit
54e6010
1 Parent(s): e840b20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -2,7 +2,7 @@ from transformers import pipeline
2
  import gradio as gr
3
  from pytube import YouTube
4
 
5
- pipe = pipeline(model = 'openai/whisper-small')
6
 
7
  def live_performance(audio):
8
  text = pipe(audio)['text']
@@ -22,8 +22,8 @@ with gr.Blocks() as demo:
22
  fn=live_performance,
23
  inputs=gr.Audio(source="microphone", type="filepath"),
24
  outputs="text",
25
- title="Whisper Small Hungarian",
26
- description="Real-time demo for Hungarian speech recognition using a fine-tuned Whisper small model."
27
  )
28
 
29
  with gr.Tab('Transcription from URL'):
@@ -31,8 +31,8 @@ with gr.Blocks() as demo:
31
  fn=url_performance,
32
  inputs=gr.Textbox(label='Paste the UL here'),
33
  outputs="text",
34
- title="Whisper Small Hungarian",
35
- description="Real-time demo for Hungarian speech recognition using a fine-tuned Whisper small model."
36
  )
37
 
38
  demo.launch()
 
2
  import gradio as gr
3
  from pytube import YouTube
4
 
5
+ pipe = pipeline(model = 'CsanadT/checkpoints')
6
 
7
  def live_performance(audio):
8
  text = pipe(audio)['text']
 
22
  fn=live_performance,
23
  inputs=gr.Audio(source="microphone", type="filepath"),
24
  outputs="text",
25
+ title="Whisper Small Swedish",
26
+ description="Real-time demo for swedish speech recognition using a fine-tuned Whisper small model."
27
  )
28
 
29
  with gr.Tab('Transcription from URL'):
 
31
  fn=url_performance,
32
  inputs=gr.Textbox(label='Paste the UL here'),
33
  outputs="text",
34
+ title="Whisper Small Swedish",
35
+ description="Real-time demo for swedish speech recognition using a fine-tuned Whisper small model."
36
  )
37
 
38
  demo.launch()