Update README.md
Browse files
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]
|
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:")
|