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

Update app.py

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