Update README.md

README.md

@@ -39,7 +39,7 @@ Developed by https://exnrt.com
 - Train Loss: 1.6578
 - Validation Loss: 1.8115
 
-You can also use `t5-small` (77M params) available in mini folder.
+You can also use `t5-small` (77M params) available in [mini](https://huggingface.co/Ateeqq/keywords-title-generator/tree/main/mini) folder.
 
 ## How to use
 
@@ -51,7 +51,7 @@ tokenizer = AutoTokenizer.from_pretrained("Ateeqq/keywords-title-generator", tok
 model = AutoModelForSeq2SeqLM.from_pretrained("Ateeqq/keywords-title-generator", token='your_token').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)
+    input_ids = tokenizer(keywords, return_tensors="pt", padding="longest", truncation=True, max_length=24).input_ids.to(device)
     outputs = model.generate(
         input_ids,
         num_beams=5,