flozi00 commited on
Commit
00c9bf5
·
1 Parent(s): 793e132

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -10,8 +10,8 @@ p = pipeline("automatic-speech-recognition", model="aware-ai/robust-wav2vec2-xls
10
  ttp = pipeline("text2text-generation", model="aware-ai/marian-german-grammar")
11
 
12
 
13
- model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B")
14
- tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_1.2B")
15
 
16
  def translate(src, tgt, text):
17
  src = src.split(" ")[-1][1:-1]
@@ -25,7 +25,7 @@ def translate(src, tgt, text):
25
  return result
26
 
27
  def transcribe(audio):
28
- transcribed = p(audio, chunk_length_s=10, stride_length_s=(4, 2))["text"]
29
 
30
  punctuated = ttp(transcribed, max_length = 512)[0]["generated_text"]
31
 
@@ -54,12 +54,12 @@ def get_translate_interface():
54
 
55
  interfaces = [
56
  get_asr_interface(),
57
- get_translate_interface(),
58
  ]
59
 
60
  names = [
61
  "ASR",
62
- "translate",
63
  ]
64
 
65
  gr.TabbedInterface(interfaces, names).launch(server_name = "0.0.0.0")
 
10
  ttp = pipeline("text2text-generation", model="aware-ai/marian-german-grammar")
11
 
12
 
13
+ #model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B")
14
+ #tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_1.2B")
15
 
16
  def translate(src, tgt, text):
17
  src = src.split(" ")[-1][1:-1]
 
25
  return result
26
 
27
  def transcribe(audio):
28
+ transcribed = p(audio, chunk_length_s=20, stride_length_s=(6, 0))["text"]
29
 
30
  punctuated = ttp(transcribed, max_length = 512)[0]["generated_text"]
31
 
 
54
 
55
  interfaces = [
56
  get_asr_interface(),
57
+ #get_translate_interface(),
58
  ]
59
 
60
  names = [
61
  "ASR",
62
+ #"translate",
63
  ]
64
 
65
  gr.TabbedInterface(interfaces, names).launch(server_name = "0.0.0.0")