Update app.py

app.py

@@ -48,7 +48,10 @@ model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M"
 
 def translate_article(article, language):
     inputs = tokenizer(article, return_tensors="pt")
-    lang_code = languages[language]
+    if language=='English':
+      lang_code='eng_Latn'
+    else:
+      lang_code = languages[language]
     translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[lang_code])
     result = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
     return result
@@ -58,7 +61,7 @@ language_choices = list(languages.keys())
 iface = gr.Interface(
     fn=translate_article,
     inputs=["text",gr.Dropdown(
-            language_choices, value="English", multiselect=False, label="Choose the language.")],
+            language_choices, value="English", multiselect=False, label="Choose the language to translate.")],
     outputs="text",
     title="Translation Tool"
 )