EricaCorral commited on
Commit
94afdd8
1 Parent(s): e27c282

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ lac = LAC(mode="seg")
12
  def make_request(chinese_text):
13
  with torch.no_grad():
14
  generated_tokens = model.generate(**tokenizer(chinese_text, return_tensors="pt", padding=True))
15
- return [tokenizer.decode(generated_tokens, skip_special_tokens=True) for t in translated]
16
 
17
  def generatepinyin(input):
18
  pinyin_list = pinyin(input)
 
12
  def make_request(chinese_text):
13
  with torch.no_grad():
14
  generated_tokens = model.generate(**tokenizer(chinese_text, return_tensors="pt", padding=True))
15
+ return [tokenizer.decode(generated_tokens, skip_special_tokens=True) for t in generated_tokens]
16
 
17
  def generatepinyin(input):
18
  pinyin_list = pinyin(input)