NeerAbhy commited on
Commit
4583afa
·
verified ·
1 Parent(s): d01bad4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -125,7 +125,7 @@ lang_id = {
125
  "Chinese": "zh",
126
  "Zulu": "zu",
127
  }
128
- def translation(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,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(translation, inputs = text0, outputs = text2)
155
  b1.click(audio_a_text, inputs=audio, outputs=text)
156
 
157
  b2 = gr.Button("Classification of speech")
158
- b2.click(text_to_sentimient, inputs=audio, outputs=text)
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()