CarlosMalaga commited on
Commit
1e2a3a3
1 Parent(s): 39aa290

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -176,7 +176,7 @@ def load_model():
176
  reader = RelikReaderForSpanExtraction("/home/user/app/models/reader",
177
  dataset_kwargs={"use_nme": True})
178
 
179
- relik = Relik(reader=reader, retriever=retriever, window_size="none", top_k=100, task="span", device="cuda", document_index_device="cpu")
180
 
181
  return relik
182
 
 
176
  reader = RelikReaderForSpanExtraction("/home/user/app/models/reader",
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