thundax commited on
Commit
b92c1d8
·
verified ·
1 Parent(s): f5bba86

Upload 3 files

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. text2sign.json +0 -0
app.py CHANGED
@@ -28,7 +28,7 @@ def do_predict(text):
28
  response_text = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
29
 
30
  signs = response_text.split(' ')
31
- actions = {x: text2sign_dict[x] or '' for x in signs}
32
 
33
  return json.dumps({'text': response_text, 'actions': actions}, ensure_ascii=False, indent=4)
34
 
 
28
  response_text = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
29
 
30
  signs = response_text.split(' ')
31
+ actions = {x: text2sign_dict.get(x, '') for x in signs}
32
 
33
  return json.dumps({'text': response_text, 'actions': actions}, ensure_ascii=False, indent=4)
34
 
text2sign.json CHANGED
The diff for this file is too large to render. See raw diff