Spaces:
Runtime error
Runtime error
Thiago Vieira
commited on
Commit
·
9b1e0fa
1
Parent(s):
0ab1893
change cache and path
Browse files
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 |
|