Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -167,8 +167,8 @@ relik_available_models = [
|
|
167 |
"relik-ie/relik-cie-small",
|
168 |
"sapienzanlp/relik-entity-linking-large",
|
169 |
"relik-ie/relik-entity-linking-large-robust",
|
|
|
170 |
"relik-ie/relik-relation-extraction-small-wikipedia-ner",
|
171 |
-
"relik-ie/relik-relation-extraction-small-wikipedia",
|
172 |
]
|
173 |
|
174 |
relik_models = {
|
@@ -205,8 +205,8 @@ relik_models = {
|
|
205 |
TaskType.TRIPLET: wikidata_index,
|
206 |
}
|
207 |
),
|
208 |
-
"relik-ie/relik-relation-extraction-small
|
209 |
-
"relik-ie/relik-relation-extraction-small
|
210 |
index=wikidata_index,
|
211 |
device="cuda",
|
212 |
retriever=wikidata_retriever,
|
|
|
167 |
"relik-ie/relik-cie-small",
|
168 |
"sapienzanlp/relik-entity-linking-large",
|
169 |
"relik-ie/relik-entity-linking-large-robust",
|
170 |
+
"relik-ie/relik-relation-extraction-small",
|
171 |
"relik-ie/relik-relation-extraction-small-wikipedia-ner",
|
|
|
172 |
]
|
173 |
|
174 |
relik_models = {
|
|
|
205 |
TaskType.TRIPLET: wikidata_index,
|
206 |
}
|
207 |
),
|
208 |
+
"relik-ie/relik-relation-extraction-small": Relik.from_pretrained(
|
209 |
+
"relik-ie/relik-relation-extraction-small",
|
210 |
index=wikidata_index,
|
211 |
device="cuda",
|
212 |
retriever=wikidata_retriever,
|