ksenia-kh commited on
Commit
0dc6062
·
1 Parent(s): 6de84bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -10,10 +10,10 @@ tokenizer.src_lang = "de"
10
  tokenizer.tgt_lang = "ca"
11
 
12
  def predict(text):
13
- encoded_ref = tokenizer(text, return_tensors="pt")
14
- generated_tokens = model.generate(**encoded_ref, forced_bos_token_id=tokenizer.get_lang_id(hyp_lang))
15
- hyp = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
16
- return hyp[0]
17
 
18
  gr.Interface(
19
  predict,
 
10
  tokenizer.tgt_lang = "ca"
11
 
12
  def predict(text):
13
+ encoded_ref = tokenizer(text, return_tensors="pt")
14
+ generated_tokens = model.generate(**encoded_ref, forced_bos_token_id=tokenizer.get_lang_id(hyp_lang))
15
+ hyp = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
16
+ return hyp[0]
17
 
18
  gr.Interface(
19
  predict,