Omnibus commited on
Commit
673c5fb
Β·
1 Parent(s): 60c2f02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -138,7 +138,8 @@ def trans_page(input,input1,trg):
138
 
139
  tokenizer.src_lang = src_lang
140
  with torch.no_grad():
141
- encoded_input = tokenizer(lang_id[], return_tensors="pt").to(device)
 
142
  generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
143
  translated_text1 = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
144
  else:
 
138
 
139
  tokenizer.src_lang = src_lang
140
  with torch.no_grad():
141
+ #lang_tr = lang_id
142
+ encoded_input = tokenizer(lang_id, return_tensors="pt").to(device)
143
  generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
144
  translated_text1 = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
145
  else: