Spaces:
Sleeping
Sleeping
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()
|