auhide commited on
Commit
aa25af6
1 Parent(s): d1e314e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -44,7 +44,7 @@ input_ids = tokenizer(input_text, return_tensors="pt").input_ids
44
  # Generate text:
45
  output = chef_gpt.generate(input_ids, max_length=150)
46
  recipe = tokenizer.batch_decode(output)[0]
47
- # Get the generated recipe - it is up until the 1st [SEP] tag.
48
  recipe = re.findall(r"\[REC\](.+?)\[SEP\]", recipe)[0]
49
 
50
  print("Съставки/Ingredients:")
 
44
  # Generate text:
45
  output = chef_gpt.generate(input_ids, max_length=150)
46
  recipe = tokenizer.batch_decode(output)[0]
47
+ # Get the generated recipe - it is up until the 1st [SEP] token.
48
  recipe = re.findall(r"\[REC\](.+?)\[SEP\]", recipe)[0]
49
 
50
  print("Съставки/Ingredients:")