jejun commited on
Commit
ca167e0
1 Parent(s): fb51173

Update demo/beam_search.py

Browse files
Files changed (1) hide show
  1. demo/beam_search.py +1 -1
demo/beam_search.py CHANGED
@@ -23,7 +23,7 @@ def skip_special_tokens_and_prettify(text, tokenizer):
23
  re.sub("|".join(tokenizer.all_special_tokens), "", text)
24
  )
25
 
26
- data = {"bitle": "", "ingredients": [], "directions": []}
27
  for section in text.split("\n"):
28
  section = section.strip()
29
  section = section.strip()
 
23
  re.sub("|".join(tokenizer.all_special_tokens), "", text)
24
  )
25
 
26
+ data = {"title": "", "ingredients": [], "directions": []}
27
  for section in text.split("\n"):
28
  section = section.strip()
29
  section = section.strip()