herme commited on
Commit
5cc0cb6
·
1 Parent(s): a8d33b3

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import whisper
3
+
4
+ # duplicate this space to run on higher CPU power for the large (very precise and multi-lingual) model to work best
5
+
6
+ def speech_to_text(uploaded, model_size):
7
+ model = whisper.load_model(model_size)
8
+ source = uploaded if uploaded is not None else ''
9
+ result = model.transcribe(source)
10
+ return f'{result["text"]}'
11
+
12
+ gr.Interface(
13
+ title="",
14
+ thumbnail="",
15
+ css="""
16
+ footer {visibility: xhidden}
17
+ .gr-prose p{text-align: center;}
18
+ .gr-button {background: black;color: white}
19
+ """,
20
+ description="",
21
+ fn=speech_to_text,
22
+ inputs=[
23
+ gr.Audio(source="upload", type="filepath", label="Upload Audio"),
24
+ gr.Dropdown(label="Select model size",value="large",choices=["tiny", "base", "small", "medium", "large"])],
25
+ outputs="text").launch(debug = True)