sappho192 commited on
Commit
0dd4a5a
·
verified ·
1 Parent(s): 7067b7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,8 +11,7 @@ model = transformers.EncoderDecoderModel.from_pretrained("sappho192/ffxiv-ja-ko-
11
  def translate(text_src):
12
  embeddings = src_tokenizer(text_src, return_attention_mask=False, return_token_type_ids=False, return_tensors='pt')
13
  embeddings = {k: v for k, v in embeddings.items()}
14
- output = model.generate(**embeddings,
15
- num_beams=5, num_return_sequences=1, no_repeat_ngram_size=1, remove_invalid_values=True)[0, 1:-1]
16
  text_trg = trg_tokenizer.decode(output.cpu())
17
  return text_trg
18
 
 
11
  def translate(text_src):
12
  embeddings = src_tokenizer(text_src, return_attention_mask=False, return_token_type_ids=False, return_tensors='pt')
13
  embeddings = {k: v for k, v in embeddings.items()}
14
+ output = model.generate(**embeddings)[0, 1:-1]
 
15
  text_trg = trg_tokenizer.decode(output.cpu())
16
  return text_trg
17