muhammadayman commited on
Commit
cbf8f16
·
1 Parent(s): 3d201bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,9 +19,9 @@ def translate_gradio(input):
19
 
20
  with tokenizer.as_target_tokenizer():
21
  input_ids = tokenizer(input, return_tensors='pt')
22
- encode = model.generate(**input_ids)
23
  # encode = model.generate(**tokenizer.prepare_seq2seq_batch(input,return_tensors='pt'))
24
- text_ar = tokenizer.batch_decode(encode,skip_special_tokens=True)[0]
25
  return text_ar
26
 
27
 
 
19
 
20
  with tokenizer.as_target_tokenizer():
21
  input_ids = tokenizer(input, return_tensors='pt')
22
+ encode = model.generate(**input_ids)
23
  # encode = model.generate(**tokenizer.prepare_seq2seq_batch(input,return_tensors='pt'))
24
+ text_ar = tokenizer.batch_decode(encode,skip_special_tokens=True)[0]
25
  return text_ar
26
 
27