Update demo/beam_search.py
Browse files- 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 = {"
|
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()
|