3ie-intervention-outcome-entity-linking

Sleeping

CarlosMalaga commited on May 6

Commit

129b641

•

1 Parent(s): bb94151

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -173,7 +173,7 @@ def load_model():
     )
     retriever.index()
-    reader = RelikReaderForSpanExtraction("/home/user/app/models/reader-extended-large",
                                         dataset_kwargs={"use_nme": True})
     relik = Relik(reader=reader, retriever=retriever, window_size="none", top_k=100, task="span", device="cpu", document_index_device="cpu")

     )
     retriever.index()
+    reader = RelikReaderForSpanExtraction("/home/user/app/models/small-extended-large-batch",
                                         dataset_kwargs={"use_nme": True})
     relik = Relik(reader=reader, retriever=retriever, window_size="none", top_k=100, task="span", device="cpu", document_index_device="cpu")