Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ header = (
|
|
45 |
"The assistant gives helpful, detailed, and polite answers to the human's questions.\n\n"
|
46 |
)
|
47 |
article = "" # insert a document here
|
48 |
-
prompt = "### Human: Please summarize the following article. {article}.\n###"
|
49 |
|
50 |
inputs = tokenizer(header + prompt, return_tensors="pt")
|
51 |
sample = model.generate(**inputs, max_length=128)
|
|
|
45 |
"The assistant gives helpful, detailed, and polite answers to the human's questions.\n\n"
|
46 |
)
|
47 |
article = "" # insert a document here
|
48 |
+
prompt = f"### Human: Please summarize the following article. {article}.\n###"
|
49 |
|
50 |
inputs = tokenizer(header + prompt, return_tensors="pt")
|
51 |
sample = model.generate(**inputs, max_length=128)
|