Update app.py

app.py

@@ -9,13 +9,13 @@ LANGUAGES = ['Akan', 'Arabic', ' Assamese', 'Bambara', 'Bengali', 'Catalan', 'En
 'Kirundi', 'Kinyarwanda', 'Shona', 'Sotho', 'Swahili', 'Tamil', 'Telugu', 'Tswana', 'Tsonga', 'Twi', 'Urdu', 'Viêt Namese', 'Wolof', 'Xhosa', 
 'Yoruba', 'Chinese', 'Zulu']
 
-API_URL = "https://api-inference.huggingface.co/models/bigscience/bloomz"
+API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
 
 
-def translate(output, text):
+def translate(input, output, text):
     """Translate text from input language to output language"""
 
-    instruction =  f"""Translatate to {output}: {text}\nTranslation: """ 
+    instruction =  f"""Translation in {input}: {text.strip()}<end> Translation in {output}:""" 
 
     json_ = {
         "inputs": instruction,
@@ -38,15 +38,16 @@ demo = gr.Blocks()
 
 with demo:
   gr.Markdown("<h1><center>Translation with Bloom</center></h1>")
-  gr.Markdown("<center>Translation in many language with mt0-xxl</center>")
+  gr.Markdown("<center>Translation with bloom.</center>")
   
   with gr.Row():
+    input_lang = gr.Dropdown(LANGUAGES, value='English', label='Select input language')
     output_lang = gr.Dropdown(LANGUAGES, value='French', label='Select output language')
 
   input_text = gr.Textbox(label="Input", lines=6)  
   output_text = gr.Textbox(lines=6, label="Output")
 
   buton = gr.Button("translate")
-  buton.click(translate, inputs=[output_lang, input_text], outputs=output_text) 
+  buton.click(translate, inputs=[input_lang, output_lang, input_text], outputs=output_text) 
     
 demo.launch(enable_queue=True, debug=True)