Update app.py

app.py

@@ -31,19 +31,16 @@ def translation(source, target, text):
     output = ""
     
     start_time = time.time()
+    chunks = text.splitlines(True)
 
-    if source = "vie_Latn" or target ="eng_Latn":
-        chunks = text.splitlines(True)
+    if source == "vie_Latn" or target == "eng_Latn":
         for chunk in chunks:
             stchunk = vinai_translator.translate_vi2en(chunk)
             output += stchunk+"\n"
-            pass
-    elif source = "eng_Latn" or target ="vie_Latn":
-        chunks = text.splitlines(True)
+    elif source == "eng_Latn" or target == "vie_Latn":
         for chunk in chunks:
             stchunk = vinai_translator.translate_en2vi(chunk)
             output += stchunk+"\n"
-            pass
     else:
         if len(model_dict) == 2:
             #model_name = 'nllb-distilled-600M'
@@ -53,11 +50,10 @@ def translation(source, target, text):
             tokenizer = model_dict[model_name + '_tokenizer']
 
             translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=source, tgt_lang=target)
-            chunks = text.splitlines(True)
             for chunk in chunks:
                 stchunk = translator(chunk+"<mask>", max_length=400, num_beams=5)
                 output += stchunk[0]['translation_text']+"\n"
-
+                
     end_time = time.time()
 
     result = {'inference_time': end_time - start_time,