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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
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}},