new commits

.gitignore

@@ -18,7 +18,6 @@ launch.json
 *.pt
 *.whl
 *.m4a
-*.csv
 input_audios/
 syn_results/
 falcon-7b-instruct/

celebbot.py

@@ -12,7 +12,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 # Build the AI
 class CelebBot():
 
-    def __init__(self, name, gender, QA_tokenizer, QA_model, sentTr_tokenizer, sentTr_model, spacy_model, knowledge_sents, top_k = 8):
+    def __init__(self, name, gender, QA_tokenizer, QA_model, sentTr_tokenizer, sentTr_model, spacy_model, knowledge_sents, top_k = 16):
         self.name = name
         self.gender = gender
         print("--- starting up", self.name, self.gender, "---")

results.csv

@@ -0,0 +1,5 @@
+model, top_k, BLEU, METEOR, ROUGEL, BERTScore F1
+flan-t5-xl, 8, 0.3, 0.45, 0.49, 0.43
+flan-t5-xl+unlimiformer, 8, 0.32, 0.46, 0.5, 0.44
+flan-t5-xl+unlimiformer, 16, 0.33, 0.47, 0.51, 0.46
+flan-t5-xl+unlimiformer, 24, 0.31, 0.45, 0.48, 0.43