don0726 commited on
Commit
9f5672d
·
verified ·
1 Parent(s): c47c6af

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import WhisperTokenizer
3
+
4
+ tokenizer = WhisperTokenizer.from_pretrained(
5
+ "openai/whisper-small"
6
+ )
7
+
8
+ def decode(text):
9
+
10
+ tokens = [
11
+ int(x.strip())
12
+ for x in text.split(",")
13
+ if x.strip()
14
+ ]
15
+
16
+ return tokenizer.decode(
17
+ tokens,
18
+ skip_special_tokens=True
19
+ )
20
+
21
+ gr.Interface(
22
+ fn=decode,
23
+
24
+ inputs=gr.Textbox(lines=8),
25
+
26
+ outputs=gr.Textbox(lines=4),
27
+
28
+ title="Whisper Token Decoder"
29
+
30
+ ).launch()