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,
|