Spaces:
Sleeping
Sleeping
Commit
·
3d201bf
1
Parent(s):
0a6fc89
Update app.py
Browse files
app.py
CHANGED
@@ -18,8 +18,8 @@ model.eval()
|
|
18 |
def translate_gradio(input):
|
19 |
|
20 |
with tokenizer.as_target_tokenizer():
|
21 |
-
|
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
|