Update test.py

test.py

@@ -9,7 +9,14 @@ from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 import gzip
 from transformers import Text2TextGenerationPipeline
-
+"""
+Works cited:
+'flan-t5-large': https://huggingface.co/google/flan-t5-large,
+'flan-t5-small': https://huggingface.co/google/flan-t5-small,
+'flan-t5-base': https://huggingface.co/google/flan-t5-base,
+'flan-t5-finetuned': finetuned version of base on the qanta guesstrain dataset,
+'flan-t5-adv-finetuned': finetuned version of finetuned base model on the adversarial dataset in qanta
+"""
 class TestEnsembleQAPipeline(Text2TextGenerationPipeline):
     def __init__(self, model=None, tokenizer=None, framework="pt", **kwargs):  
         super().__init__(model=model, tokenizer=tokenizer, framework=framework)