Added language selector

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 from PIL import Image
+from lang_list import LANG
 
 ##Image Classification
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
@@ -20,8 +21,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 modelt = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
 tokenizert = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 
-def translation(text):
-  translator = pipeline('translation', model=modelt, tokenizer=tokenizert, src_lang="eng_Latn", tgt_lang='aeb_Arab')
+def translation(text,target):
+  translator = pipeline('translation', model=modelt, tokenizer=tokenizert, src_lang="eng_Latn", tgt_lang=target)
   output = translator(text)
   return (output[0]['translation_text'])
 
@@ -32,7 +33,8 @@ with demo:
     b1 = gr.Button("Recognize Image")
     text = gr.Textbox()
     b1.click(image_to_text, inputs=image_file, outputs=text)
+    target = gr.Dropdown(LANG,interactive=True,label="Target Language")
     b2 = gr.Button("Translation")
     out1 = gr.Textbox()
-    b2.click(translation, inputs=text, outputs=out1)
+    b2.click(translation, inputs=[text,target], outputs=out1)
 demo.launch()