Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 # Use a pipeline as a high-level helper
 from transformers import pipeline
 
-pipe = pipeline(
+text_translator = pipeline(
     "translation", 
     model="facebook/nllb-200-distilled-600M",
     torch_dtype=torch.bfloat16)
@@ -22,11 +22,11 @@ def get_FLORES_code_from_language(language):
 
 
 def translate_text(text, destination_language):
-    # text = "Hello Friends, How are you?"
-    dest_code= get_FLORES_code_from_language(destination_language)
-    translation = text_translator(text,
-                                  src_lang="eng_Latn",
-                                  tgt_lang=dest_code)
+    dest_code = get_FLORES_code_from_language(destination_language)
+    if dest_code is None:
+        return "Invalid destination language."
+
+    translation = text_translator(text, src_lang="eng_Latn", tgt_lang=dest_code)
     return translation[0]["translation_text"]
 
 gr.close_all()