jslin09 commited on
Commit
1d262ca
·
verified ·
1 Parent(s): 16fabf5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -43,7 +43,8 @@ def generate(prompt):
43
  top_k=50,
44
  top_p=0.9
45
  )
46
- return tokenizer.decode(results[0])
 
47
 
48
  examples = [
49
  ["闕很大明知金融帳戶之存摺、提款卡及密碼係供自己使用之重要理財工具,"],
 
43
  top_k=50,
44
  top_p=0.9
45
  )
46
+ result = tokenizer.decode(results[0]).split("<bos>")[1]
47
+ return result
48
 
49
  examples = [
50
  ["闕很大明知金融帳戶之存摺、提款卡及密碼係供自己使用之重要理財工具,"],