s3nh commited on
Commit
872f66f
·
1 Parent(s): 15d35d6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  Buy me a coffee if you like this project ;)
2
  <a href="https://www.buymeacoffee.com/s3nh"><img src="https://www.buymeacoffee.com/assets/img/guidelines/download-assets-sm-1.svg" alt=""></a>
3
 
@@ -54,4 +57,4 @@ encoded_new = tokenizer_new(q, return_tensors = "pt")["input_ids"]
54
  generated_new = model_new.generate(encoded_new.cuda())[0, encoded_new.shape[-1]:]
55
  decoded_new = tokenizer_new.decode(generated_new, skip_special_tokens=True).strip()
56
  pprint(decoded_new)
57
- ```
 
1
+ ---
2
+ pipeline_tag: text-generation
3
+ ---
4
  Buy me a coffee if you like this project ;)
5
  <a href="https://www.buymeacoffee.com/s3nh"><img src="https://www.buymeacoffee.com/assets/img/guidelines/download-assets-sm-1.svg" alt=""></a>
6
 
 
57
  generated_new = model_new.generate(encoded_new.cuda())[0, encoded_new.shape[-1]:]
58
  decoded_new = tokenizer_new.decode(generated_new, skip_special_tokens=True).strip()
59
  pprint(decoded_new)
60
+ ```