relik-ie commited on
Commit
4712226
·
verified ·
1 Parent(s): b6816e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
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
- "relik-ie/relik-relation-extraction-large-wikipedia",
134
  "relik-ie/relik-entity-linking-large-robust",
 
135
  ]
136
- # demo_index_el = {
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",