Update app.py

app.py

@@ -175,8 +175,12 @@ relik_models = {
     "sapienzanlp/relik-entity-linking-large": Relik.from_pretrained(
         "sapienzanlp/relik-entity-linking-large",
         device="cuda",
-        index=wikipedia_index,
-        retriever=wikipedia_retriever,
+        index={
+            TaskType.SPAN: wikipedia_index,
+        },
+        retriever={
+            TaskType.SPAN:wikipedia_retriever,
+        },
         reader_kwargs={"dataset_kwargs": {"use_nme": True}},
     ),
     "relik-ie/relik-cie-small": Relik.from_pretrained(
@@ -207,15 +211,23 @@ relik_models = {
     ),
     "relik-ie/relik-relation-extraction-small": Relik.from_pretrained(
         "relik-ie/relik-relation-extraction-small",
-        index=wikidata_index,
+        index={
+            TaskType.TRIPLET:wikidata_index,
+        },
         device="cuda",
-        retriever=wikidata_retriever,
+        retriever={
+            TaskType.TRIPLET: wikidata_retriever,
+        },
     ),
     "relik-ie/relik-entity-linking-large-robust": Relik.from_pretrained(
         "relik-ie/relik-entity-linking-large-robust",
-        index=wikipedia_index,
+        index={
+            TaskType.SPAN: wikipedia_index,
+        },
         device="cuda",
-        retriever=wikipedia_retriever,
+        retriever={
+            TaskType.SPAN: wikipedia_retriever,
+        },
         reader_kwargs={"dataset_kwargs": {"use_nme": True}},
     ),
 }