Spaces:

minhdang
/

code

Paused

minhdang commited on Mar 19

Commit

ddae4cd

•

1 Parent(s): 31c25ec

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,7 +73,7 @@ model_vi2en = AutoModelForSeq2SeqLM.from_pretrained("vinai/vinai-translate-vi2en
 def translate_vi2en(vi_text: str) -> str:
     for i, j in dict_map.items():
         vi_text = vi_text.replace(i, j)
-    input_ids = tokenizer_vi2en(vi_text, return_tensors="pt")to("cuda").input_ids
     output_ids = model_vi2en.generate(
         input_ids,
         decoder_start_token_id=tokenizer_vi2en.lang_code_to_id["en_XX"],

 def translate_vi2en(vi_text: str) -> str:
     for i, j in dict_map.items():
         vi_text = vi_text.replace(i, j)
+    input_ids = tokenizer_vi2en(vi_text, return_tensors="pt").to("cuda").input_ids
     output_ids = model_vi2en.generate(
         input_ids,
         decoder_start_token_id=tokenizer_vi2en.lang_code_to_id["en_XX"],