Update app.py

app.py

@@ -36,8 +36,32 @@ def translate(text):
 
     outs = model.generate(**{**ins, **gen_args})
     output = tokenizer.batch_decode(outs.sequences, skip_special_tokens=True)
+    text2 = '\n'.join(output)
+    
+    ##################
+
+    tokenizer.src_lang = "tl"
+    tokenizer.tgt_lang = "en"
+
+    ins = tokenizer(text2, return_tensors='pt').to(device)
+
+    gen_args = {
+            'return_dict_in_generate': True,
+            'output_scores': True,
+            'output_hidden_states': True,
+            'length_penalty': 0.0,  # don't encourage longer or shorter output,
+            'num_return_sequences': 1,
+            'num_beams':1,
+            'forced_bos_token_id': tokenizer.lang_code_to_id["en"]
+        }
+    
+
+    outs2 = model.generate(**{**ins, **gen_args})
+    output2 = tokenizer.batch_decode(outs2.sequences, skip_special_tokens=True)
+
+
 
-    return '\n'.join(output)
+    return '\n'.join(output2)
 
 with gr.Blocks() as app:
     markdown="""