Update README.md

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 tags:
 - medical
+datasets:
+- allenai/s2orc
 ---
 This repo contains PMC_LLaMA_7B, which is LLaMA-7b finetuned on the PMC papers in S2ORC dataset.
 
@@ -30,6 +32,4 @@ batch = tokenizer(
 with torch.no_grad():
     generated = model.generate(inputs = batch["input_ids"], max_length=200, do_sample=True, top_k=50)
     print('model predict: ',tokenizer.decode(generated[0]))
-```
-
-
+```