CarlosMalaga commited on
Commit
f2b8939
1 Parent(s): 129b641

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, window_size="none", top_k=100, task="span", device="cpu", document_index_device="cpu")
180
 
181
  return relik
182
 
@@ -271,7 +271,7 @@ def run_client():
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
 
 
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