Update README.md
Browse files
README.md
CHANGED
@@ -48,7 +48,7 @@ We use the prompt template `Publisher: {vox} article: ` for training. We trained
|
|
48 |
|
49 |
>>> publisher = "Reuters"
|
50 |
>>> assert publisher in ["Reuters", "NYT", "CNBC", "Hill", "People", "CNN", "Vice", "Mashable", "Refinery", "BI", "TechCrunch", "Verge", "TMZ", "Axios", "Vox", "Guardian", "BBCNews", "WashingtonPost", "USAToday"]
|
51 |
-
>>> prompt = f"Publisher: {publisher.lower()} article: Local police is dealing with a car accident"
|
52 |
|
53 |
>>> inputs = tokenizer(prompt, return_tensors="pt")
|
54 |
>>> out = model.generate(**inputs, penalty_alpha=0.6)
|
|
|
48 |
|
49 |
>>> publisher = "Reuters"
|
50 |
>>> assert publisher in ["Reuters", "NYT", "CNBC", "Hill", "People", "CNN", "Vice", "Mashable", "Refinery", "BI", "TechCrunch", "Verge", "TMZ", "Axios", "Vox", "Guardian", "BBCNews", "WashingtonPost", "USAToday"]
|
51 |
+
>>> prompt = f"{tokenizer.bos_token}Publisher: {publisher.lower()} article: Local police is dealing with a car accident"
|
52 |
|
53 |
>>> inputs = tokenizer(prompt, return_tensors="pt")
|
54 |
>>> out = model.generate(**inputs, penalty_alpha=0.6)
|