flozi00 commited on
Commit
7226ea6
·
1 Parent(s): ea344d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -3
app.py CHANGED
@@ -9,10 +9,12 @@ ttp = pipeline("text2text-generation", model="aware-ai/marian-german-grammar")
9
 
10
  def transcribe(audio):
11
  transcribed = p(audio, chunk_length_s=16, stride_length_s=(4, 0))["text"]
 
 
12
 
 
13
  punctuated = ttp(transcribed, max_length = 512)[0]["generated_text"]
14
-
15
- return transcribed, punctuated
16
 
17
  def get_asr_interface():
18
  return gr.Interface(
@@ -22,15 +24,26 @@ def get_asr_interface():
22
  ],
23
  outputs=[
24
  "textbox",
25
- "textbox"
 
 
 
 
 
 
 
 
 
26
  ])
27
 
28
  interfaces = [
29
  get_asr_interface(),
 
30
  ]
31
 
32
  names = [
33
  "ASR",
 
34
  ]
35
 
36
  gr.TabbedInterface(interfaces, names).launch(server_name = "0.0.0.0", enable_queue=False)
 
9
 
10
  def transcribe(audio):
11
  transcribed = p(audio, chunk_length_s=16, stride_length_s=(4, 0))["text"]
12
+
13
+ return transcribed
14
 
15
+ def punctuate(transcribed):
16
  punctuated = ttp(transcribed, max_length = 512)[0]["generated_text"]
17
+ return punctuated
 
18
 
19
  def get_asr_interface():
20
  return gr.Interface(
 
24
  ],
25
  outputs=[
26
  "textbox",
27
+ ])
28
+
29
+ def get_punctuation_interface():
30
+ return gr.Interface(
31
+ fn=punctuate,
32
+ inputs=[
33
+ "textbox",
34
+ ],
35
+ outputs=[
36
+ "textbox",
37
  ])
38
 
39
  interfaces = [
40
  get_asr_interface(),
41
+ get_punctuation_interface(),
42
  ]
43
 
44
  names = [
45
  "ASR",
46
+ "GRAMMAR",
47
  ]
48
 
49
  gr.TabbedInterface(interfaces, names).launch(server_name = "0.0.0.0", enable_queue=False)