Update README.md

README.md

@@ -16,6 +16,8 @@ You can use this model for headline generation task on English news articles.
 
 ### Usage
 ```python
+
+
 article = """Two of the OPEC oil cartel’s 11 members, Nigeria and Venezuela, said today that they would voluntarily cut production in response to declining crude oil prices, which have fallen 20 percent from their peak two months ago.
 
 The move, which would take less than 200,000 barrels of oil a day off the market, follows days of mixed signals from some OPEC officials, who have voiced increasing concern about the rapid drop in prices. Nigeria’s oil minister, Edmund Daukoru, who is president of OPEC this year, recently said the price of oil was “very low.”
@@ -48,6 +50,8 @@ from transformers import AutoModelWithLMHead,AutoModelForSeq2SeqLM, DataCollator
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(device)
 
+#appending the task identifier to the beginning of input
+article = "Headline: " + article
 
 model = AutoModelForSeq2SeqLM.from_pretrained("omidvaramin/Hprophetnet-large").to(device)
 tokenizer = AutoTokenizer.from_pretrained("omidvaramin/Hprophetnet-large")