nltk error addressed

app.py

@@ -3,6 +3,8 @@ import os
 from transformers import AutoTokenizer, AutoModel
 from sentence_transformers import SentenceTransformer
 import pickle
+import nltk
+nltk.download('punkt')
 
 from input_format import *
 from score import *
@@ -193,4 +195,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch() 
+    demo.launch() 

score.py

@@ -146,4 +146,4 @@ def compute_overall_score(doc_model, tokenizer, query, papers, batch=5):
     abstracts_sorted = [abstracts[x] for x in idx_sorted]
     scores_sorted = [scores[x] for x in idx_sorted]
     
-    return titles_sorted, abstracts_sorted, scores_sorted
+    return titles_sorted, abstracts_sorted, scores_sorted