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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ model.eval()
18
  def translate_gradio(input):
19
 
20
  with tokenizer.as_target_tokenizer():
21
- encode = model.generate(tokenizer(input, return_tensors='pt'))
22
-
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
 
18
  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