Update README.md

README.md

@@ -1,7 +1,7 @@
 ---
 datasets:
 - hotpot_qa
-- canard
+- gaussalgo/Canard_Wiki-augmented
 ---
 
 # Model Card for T5-LM-Large_Canard-HotpotQA-rephrase 
@@ -15,8 +15,8 @@ The best checkpoint was picked by a minimal loss on all (3) training objectives.
 
 ```python
 import datasets
-canard_train_augm = datasets.load_from_disk("canard_train_augm_full.hf")  # constructed in notebook: 2.1_construct_qa_dataset.ipynb
-canard_test_augm = datasets.load_from_disk("canard_test_augm_full.hf")
+canard_train_augm = datasets.load_dataset("gaussalgo/Canard_Wiki-augmented", split="train")  # see the dataset card for details
+canard_test_augm = datasets.load_dataset("gaussalgo/Canard_Wiki-augmented", split="test")
 
 canard_df = canard_train_augm.to_pandas()
 canard_test_df = canard_train_augm.to_pandas()
@@ -105,6 +105,9 @@ rephrasing_val_labels = canard_test_df.Rewrite.values
 print(rephrasing_labels[0])
 
 # Training
+# see Adaptor's homepage for details:
+# https://github.com/gaussalgo/adaptor
+
 from adaptor.lang_module import LangModule
 
 lang_module = LangModule("google/t5-large-lm-adapt")
@@ -191,4 +194,4 @@ See the prompting templates used in training to infer the optimal prompting form
 
 #### Contact
 
-Feel free to ask questions at stefanik{at} gaussalgo.com
+Feel free to ask questions here, or at stefanik{at} gaussalgo.com