Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -128,17 +128,12 @@ wikidata_retriever = GoldenRetriever("relik-ie/encoder-e5-small-v2-wikipedia-rel
|
|
128 |
wikidata_index = InMemoryDocumentIndex.from_pretrained("relik-ie/encoder-e5-small-v2-wikipedia-relations-index")
|
129 |
|
130 |
relik_available_models = [
|
131 |
-
"sapienzanlp/relik-entity-linking-large",
|
132 |
"relik-ie/relik-reader-small-cie-wikipedia",
|
133 |
-
"
|
134 |
"relik-ie/relik-entity-linking-large-robust",
|
|
|
135 |
]
|
136 |
-
|
137 |
-
# "span": {
|
138 |
-
# "_target_": "relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained",
|
139 |
-
# "name_or_path": "sapienzanlp/relik-retriever-e5-base-v2-aida-blink-wikipedia-2Mpopular-index",
|
140 |
-
# }
|
141 |
-
# }
|
142 |
relik_models = {
|
143 |
"sapienzanlp/relik-entity-linking-large": Relik.from_pretrained(
|
144 |
"sapienzanlp/relik-entity-linking-large",
|
|
|
128 |
wikidata_index = InMemoryDocumentIndex.from_pretrained("relik-ie/encoder-e5-small-v2-wikipedia-relations-index")
|
129 |
|
130 |
relik_available_models = [
|
|
|
131 |
"relik-ie/relik-reader-small-cie-wikipedia",
|
132 |
+
"sapienzanlp/relik-entity-linking-large",
|
133 |
"relik-ie/relik-entity-linking-large-robust",
|
134 |
+
"relik-ie/relik-relation-extraction-large-wikipedia",
|
135 |
]
|
136 |
+
|
|
|
|
|
|
|
|
|
|
|
137 |
relik_models = {
|
138 |
"sapienzanlp/relik-entity-linking-large": Relik.from_pretrained(
|
139 |
"sapienzanlp/relik-entity-linking-large",
|