Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -125,7 +125,7 @@ lang_id = {
|
|
125 |
"Chinese": "zh",
|
126 |
"Zulu": "zu",
|
127 |
}
|
128 |
-
def
|
129 |
src_lang = lang_id[source_lang]
|
130 |
trg_lang = lang_id[target_lang]
|
131 |
tokenizer.src_lang = src_lang
|
@@ -135,6 +135,9 @@ def translation(source_lang, target_lang):
|
|
135 |
translated_text = tokenizer.batch_decode(
|
136 |
generated_tokens, skip_special_tokens=True)[0]
|
137 |
return translated_text
|
|
|
|
|
|
|
138 |
demo = gr.Blocks()
|
139 |
|
140 |
with demo:
|
@@ -151,10 +154,10 @@ with demo:
|
|
151 |
# source_lang])
|
152 |
b1 = gr.Button("convert to text")
|
153 |
b3 = gr.Button("translate")
|
154 |
-
b3.click(
|
155 |
b1.click(audio_a_text, inputs=audio, outputs=text)
|
156 |
|
157 |
b2 = gr.Button("Classification of speech")
|
158 |
-
b2.click(
|
159 |
|
160 |
demo.launch()
|
|
|
125 |
"Chinese": "zh",
|
126 |
"Zulu": "zu",
|
127 |
}
|
128 |
+
def translation_text(source_lang, target_lang):
|
129 |
src_lang = lang_id[source_lang]
|
130 |
trg_lang = lang_id[target_lang]
|
131 |
tokenizer.src_lang = src_lang
|
|
|
135 |
translated_text = tokenizer.batch_decode(
|
136 |
generated_tokens, skip_special_tokens=True)[0]
|
137 |
return translated_text
|
138 |
+
|
139 |
+
def print_s(arr):
|
140 |
+
return arr[0], arr[1]
|
141 |
demo = gr.Blocks()
|
142 |
|
143 |
with demo:
|
|
|
154 |
# source_lang])
|
155 |
b1 = gr.Button("convert to text")
|
156 |
b3 = gr.Button("translate")
|
157 |
+
b3.click(translation_text, inputs = text0, outputs = text2)
|
158 |
b1.click(audio_a_text, inputs=audio, outputs=text)
|
159 |
|
160 |
b2 = gr.Button("Classification of speech")
|
161 |
+
b2.click(print_s, inputs=[source_lang, target_lang], outputs=text)
|
162 |
|
163 |
demo.launch()
|