Omnibus commited on
Commit
60c2f02
Β·
1 Parent(s): 6f0b323

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -138,13 +138,13 @@ def trans_page(input,input1,trg):
138
 
139
  tokenizer.src_lang = src_lang
140
  with torch.no_grad():
141
- encoded_input = tokenizer(input1, 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:
145
  translated_text1=input1
146
  pass
147
- return translated_text,translated_text1
148
 
149
 
150
  def trans_to(input,src,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:
145
  translated_text1=input1
146
  pass
147
+ return translated_text,gr.Dropdown.update(choices=list(translated_text1.keys()))
148
 
149
 
150
  def trans_to(input,src,trg):