letrunglinh commited on
Commit
fef4425
1 Parent(s): a369c76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -5,12 +5,15 @@ from qa_model import *
5
  from bm25_utils import *
6
  from pairwise_model import *
7
 
 
 
 
 
8
  df_wiki_windows = pd.read_csv("./processed/wikipedia_chungta_cleaned.csv")
9
  df_wiki = pd.read_csv("./processed/wikipedia_chungta_short.csv")
10
  df_wiki.title = df_wiki.title.apply(str)
11
 
12
 
13
-
14
  entity_dict = json.load(open("./processed/entities.json"))
15
  new_dict = dict()
16
  for key, val in entity_dict.items():
 
5
  from bm25_utils import *
6
  from pairwise_model import *
7
 
8
+ import nltk
9
+ nltk.download('punkt')
10
+
11
+
12
  df_wiki_windows = pd.read_csv("./processed/wikipedia_chungta_cleaned.csv")
13
  df_wiki = pd.read_csv("./processed/wikipedia_chungta_short.csv")
14
  df_wiki.title = df_wiki.title.apply(str)
15
 
16
 
 
17
  entity_dict = json.load(open("./processed/entities.json"))
18
  new_dict = dict()
19
  for key, val in entity_dict.items():