CarlosMalaga
commited on
Commit
•
f2b8939
1
Parent(s):
129b641
Update app.py
Browse files
app.py
CHANGED
@@ -176,7 +176,7 @@ def load_model():
|
|
176 |
reader = RelikReaderForSpanExtraction("/home/user/app/models/small-extended-large-batch",
|
177 |
dataset_kwargs={"use_nme": True})
|
178 |
|
179 |
-
relik = Relik(reader=reader, retriever=retriever,
|
180 |
|
181 |
return relik
|
182 |
|
@@ -271,7 +271,7 @@ def run_client():
|
|
271 |
## Possible Candidates:
|
272 |
- {}
|
273 |
</span>
|
274 |
-
""".format("
|
275 |
|
276 |
st.markdown(text, unsafe_allow_html=True)
|
277 |
|
|
|
176 |
reader = RelikReaderForSpanExtraction("/home/user/app/models/small-extended-large-batch",
|
177 |
dataset_kwargs={"use_nme": True})
|
178 |
|
179 |
+
relik = Relik(reader=reader, retriever=retriever, window_size=32, window_stride=16, top_k=100, task="span", device="cpu", document_index_device="cpu")
|
180 |
|
181 |
return relik
|
182 |
|
|
|
271 |
## Possible Candidates:
|
272 |
- {}
|
273 |
</span>
|
274 |
+
""".format("\\n- ".join(dict_of_ents_candidates["ents"][2:12]))
|
275 |
|
276 |
st.markdown(text, unsafe_allow_html=True)
|
277 |
|