Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -122,10 +122,10 @@ For more information, please refer to the [source code](https://github.com/Sapie
|
|
122 |
"""
|
123 |
|
124 |
wikipedia_retriever = GoldenRetriever("relik-ie/encoder-e5-base-v2-wikipedia")
|
125 |
-
wikipedia_index = InMemoryDocumentIndex.from_pretrained("relik-ie/encoder-e5-base-v2-wikipedia-index")
|
126 |
|
127 |
wikidata_retriever = GoldenRetriever("relik-ie/encoder-e5-small-v2-wikipedia-relations")
|
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",
|
@@ -137,14 +137,12 @@ relik_available_models = [
|
|
137 |
relik_models = {
|
138 |
"sapienzanlp/relik-entity-linking-large": Relik.from_pretrained(
|
139 |
"sapienzanlp/relik-entity-linking-large",
|
140 |
-
index_precision="bf16",
|
141 |
index=wikipedia_index,
|
142 |
retriever=wikipedia_retriever,
|
143 |
reader_kwargs={"dataset_kwargs": {"use_nme": True}},
|
144 |
),
|
145 |
"relik-ie/relik-reader-small-cie-wikipedia": Relik.from_pretrained(
|
146 |
"relik-ie/relik-reader-small-cie-wikipedia",
|
147 |
-
index_precision="bf16",
|
148 |
# reader="/media/data/relik/models/PereLluis13/relik-reader-deberta-large-wikipedia-aida-full-interleave",
|
149 |
device="cuda",
|
150 |
index={
|
@@ -159,14 +157,12 @@ relik_models = {
|
|
159 |
),
|
160 |
"relik-ie/relik-relation-extraction-large-wikipedia": Relik.from_pretrained(
|
161 |
"relik-ie/relik-relation-extraction-large-wikipedia",
|
162 |
-
index_precision="bf16",
|
163 |
index=wikidata_index,
|
164 |
retriever=wikidata_retriever,
|
165 |
device="cuda",
|
166 |
),
|
167 |
"relik-ie/relik-entity-linking-large-robust": Relik.from_pretrained(
|
168 |
"relik-ie/relik-entity-linking-large-robust",
|
169 |
-
index_precision="bf16",
|
170 |
index=wikipedia_index,
|
171 |
retriever=wikipedia_retriever,
|
172 |
reader_kwargs={"dataset_kwargs": {"use_nme": True}},
|
|
|
122 |
"""
|
123 |
|
124 |
wikipedia_retriever = GoldenRetriever("relik-ie/encoder-e5-base-v2-wikipedia")
|
125 |
+
wikipedia_index = InMemoryDocumentIndex.from_pretrained("relik-ie/encoder-e5-base-v2-wikipedia-index", index_precision="bf16")
|
126 |
|
127 |
wikidata_retriever = GoldenRetriever("relik-ie/encoder-e5-small-v2-wikipedia-relations")
|
128 |
+
wikidata_index = InMemoryDocumentIndex.from_pretrained("relik-ie/encoder-e5-small-v2-wikipedia-relations-index", index_precision="bf16")
|
129 |
|
130 |
relik_available_models = [
|
131 |
"relik-ie/relik-reader-small-cie-wikipedia",
|
|
|
137 |
relik_models = {
|
138 |
"sapienzanlp/relik-entity-linking-large": Relik.from_pretrained(
|
139 |
"sapienzanlp/relik-entity-linking-large",
|
|
|
140 |
index=wikipedia_index,
|
141 |
retriever=wikipedia_retriever,
|
142 |
reader_kwargs={"dataset_kwargs": {"use_nme": True}},
|
143 |
),
|
144 |
"relik-ie/relik-reader-small-cie-wikipedia": Relik.from_pretrained(
|
145 |
"relik-ie/relik-reader-small-cie-wikipedia",
|
|
|
146 |
# reader="/media/data/relik/models/PereLluis13/relik-reader-deberta-large-wikipedia-aida-full-interleave",
|
147 |
device="cuda",
|
148 |
index={
|
|
|
157 |
),
|
158 |
"relik-ie/relik-relation-extraction-large-wikipedia": Relik.from_pretrained(
|
159 |
"relik-ie/relik-relation-extraction-large-wikipedia",
|
|
|
160 |
index=wikidata_index,
|
161 |
retriever=wikidata_retriever,
|
162 |
device="cuda",
|
163 |
),
|
164 |
"relik-ie/relik-entity-linking-large-robust": Relik.from_pretrained(
|
165 |
"relik-ie/relik-entity-linking-large-robust",
|
|
|
166 |
index=wikipedia_index,
|
167 |
retriever=wikipedia_retriever,
|
168 |
reader_kwargs={"dataset_kwargs": {"use_nme": True}},
|