relik-ie commited on
Commit
11d8b13
·
verified ·
1 Parent(s): 64850ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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-wikipedia": Relik.from_pretrained(
209
- "relik-ie/relik-relation-extraction-small-wikipedia",
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,