Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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")
|