Thiago Vieira commited on
Commit
9b1e0fa
·
1 Parent(s): 0ab1893

change cache and path

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -140,7 +140,7 @@ def load_docs():
140
  doc_set = {}
141
  doc_id = ""
142
  doc_text = ""
143
- with open('content/CISI.ALL') as f:
144
  lines = ""
145
  for l in f.readlines():
146
  lines += "\n" + l.strip() if l.startswith(".") else " " + l.strip()
@@ -160,15 +160,15 @@ def load_docs():
160
 
161
  @st.cache_resource
162
  def load_models():
163
- with open('models/BM25_simple.pkl', 'rb') as file:
164
  bm25_simple: BM25Simple = pickle.load(file)
165
  print(bm25_simple.corpus_size)
166
 
167
- with open('models/BM25OKapi.pkl', 'rb') as file:
168
  bm25_okapi: BM25Okapi = pickle.load(file)
169
  print(bm25_okapi.corpus_size)
170
 
171
- with open('models/BM25Plus.pkl', 'rb') as file:
172
  bm25_plus: BM25Plus = pickle.load(file)
173
  print(bm25_plus.corpus_size)
174
 
 
140
  doc_set = {}
141
  doc_id = ""
142
  doc_text = ""
143
+ with open(path + '/content/CISI.ALL') as f:
144
  lines = ""
145
  for l in f.readlines():
146
  lines += "\n" + l.strip() if l.startswith(".") else " " + l.strip()
 
160
 
161
  @st.cache_resource
162
  def load_models():
163
+ with open(path + '/models/BM25_simple.pkl', 'rb') as file:
164
  bm25_simple: BM25Simple = pickle.load(file)
165
  print(bm25_simple.corpus_size)
166
 
167
+ with open(path + '/models/BM25OKapi.pkl', 'rb') as file:
168
  bm25_okapi: BM25Okapi = pickle.load(file)
169
  print(bm25_okapi.corpus_size)
170
 
171
+ with open(path + '/models/BM25Plus.pkl', 'rb') as file:
172
  bm25_plus: BM25Plus = pickle.load(file)
173
  print(bm25_plus.corpus_size)
174