h4d35 commited on
Commit
e258488
1 Parent(s): 62ca4f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,13 +3,13 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  model_translate = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M")
5
  tokenizer_translate = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
6
- inlang='en'
7
- outlang='hi'
8
  tokenizer_translate.src_lang = inlang
9
 
10
  def translate(text):
11
- encoded_en = tokenizer_translate(text, return_tensors="pt")
12
- generated_tokens = model_translate.generate(**encoded_en, forced_bos_token_id=tokenizer_translate.get_lang_id(outlang))
13
  return tokenizer_translate.batch_decode(generated_tokens, skip_special_tokens=True)[0]
14
 
15
  interface = gr.Interface(
 
3
 
4
  model_translate = AutoModelForSeq2SeqLM.from_pretrained("facebook/m2m100_418M")
5
  tokenizer_translate = AutoTokenizer.from_pretrained("facebook/m2m100_418M")
6
+ inlang='hi'
7
+ outlang='en'
8
  tokenizer_translate.src_lang = inlang
9
 
10
  def translate(text):
11
+ encoded_hi = tokenizer_translate(text, return_tensors="pt")
12
+ generated_tokens = model_translate.generate(**encoded_hi, forced_bos_token_id=tokenizer_translate.get_lang_id(outlang))
13
  return tokenizer_translate.batch_decode(generated_tokens, skip_special_tokens=True)[0]
14
 
15
  interface = gr.Interface(