Remove low quality languages from target languages (#5)

- Remove low quality languages from target languages (49900d0cf969370e83745811c86ce1d8ed17b1db)

app.py

@@ -8,12 +8,15 @@ import nltk
 
 nltk.download("punkt")
 
+REMOVED_TARGET_LANGUAGES = {"Ligurian", "Lombard", "Sicilian"}
+
+
 device = "cpu" if platform.system() == "Darwin" else "cuda"
 MODEL_NAME = "facebook/nllb-200-3.3B"
 
 code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[1]))
 flores_codes = list(code_mapping.keys())
-
+target_languages = [language for language in flores_codes if not language in REMOVED_TARGET_LANGUAGES]
 
 def load_model():
     model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
@@ -77,7 +80,7 @@ with gr.Blocks() as demo:
     gr.Markdown(description)
     with gr.Row():
         src_lang = gr.Dropdown(label="Source Language", choices=flores_codes)
-        target_lang = gr.Dropdown(label="Target Language", choices=flores_codes)
+        target_lang = gr.Dropdown(label="Target Language", choices=target_languages)
     with gr.Row():
         input_text = gr.Textbox(label="Input Text", lines=6)
     with gr.Row():