Update README.md

README.md

@@ -50,8 +50,8 @@ device = "cuda"
 tokenizer = AutoTokenizer.from_pretrained("Ateeqq/keywords-title-generator", token='your_token')
 model = AutoModelForSeq2SeqLM.from_pretrained("Ateeqq/keywords-title-generator", token='your_token').to(device)
 
-def generate_title(question):
-    input_ids = tokenizer(f'generate title: {question}', return_tensors="pt", padding="longest", truncation=True, max_length=24).input_ids.to(device)
+def generate_title(keywords):
+    input_ids = tokenizer(f'generate title: {keywords}', return_tensors="pt", padding="longest", truncation=True, max_length=24).input_ids.to(device)
     outputs = model.generate(
         input_ids,
         num_beams=5,
@@ -65,8 +65,8 @@ def generate_title(question):
     )
     return tokenizer.batch_decode(outputs, skip_special_tokens=True)
 
-text = 'book, history, kids'
-generate_title(text)
+keywords = 'book, history, kids'
+generate_title(keywords)
 ```
 ### Output:
 ```