CarlosMalaga
commited on
Commit
•
1e2a3a3
1
Parent(s):
39aa290
Update app.py
Browse files
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="
|
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 |
|