Update README.md

README.md

@@ -1,5 +1,25 @@
 ---
 license: mit
+datasets:
+- monsoon-nlp/greenbeing-proteins
 ---
 
-Adapter model / weights only for https://huggingface.co/monsoon-nlp/nyc-savvy-llama2-7b
+Adapter model / weights only for https://huggingface.co/monsoon-nlp/nyc-savvy-llama2-7b
+
+## Usage
+
+```
+from peft import AutoPeftModelForCausalLM
+from transformers import AutoTokenizer
+
+# this model
+model = AutoPeftModelForCausalLM.from_pretrained("monsoon-nlp/tinyllama-mixpretrain-uniprottune").to("cuda")
+# base model for the tokenizer
+tokenizer = AutoTokenizer.from_pretrained("monsoon-nlp/tinyllama-mixpretrain-quinoa-sciphi")
+
+inputs = tokenizer("<sequence> Subcellular locations:", return_tensors="pt")
+outputs = model.generate(input_ids=inputs["input_ids"].to("cuda"), max_new_tokens=50)
+print(tokenizer.batch_decode(outputs, skip_special_tokens=True)[0])
+```
+
+Notebook: https://colab.research.google.com/drive/1UTavcVpqWkp4C_GkkS_HxDQ0Orpw43iu?usp=sharing