Update app.py

app.py

@@ -165,9 +165,11 @@ ner_type_retriever = GoldenSillyRetriever(
 
 relik_available_models = [
     "relik-ie/relik-cie-small",
+    "relik-ie/relik-cie-xl",
     "sapienzanlp/relik-entity-linking-large",
     "relik-ie/relik-entity-linking-large-robust",
     "relik-ie/relik-relation-extraction-small",
+    "relik-ie/relik-relation-extraction-large",
     "relik-ie/relik-relation-extraction-small-wikipedia-ner",
 ]
 
@@ -196,6 +198,19 @@ relik_models = {
         },
         reader_kwargs={"dataset_kwargs": {"use_nme": True}},
     ),
+    "relik-ie/relik-cie-xl": Relik.from_pretrained(
+        "relik-ie/relik-cie-xl",
+        device="cuda",
+        index={
+            TaskType.SPAN: wikipedia_index,
+            TaskType.TRIPLET: wikidata_index,
+        },
+        retriever={
+            TaskType.SPAN: wikipedia_retriever,
+            TaskType.TRIPLET: wikidata_retriever,
+        },
+        reader_kwargs={"dataset_kwargs": {"use_nme": True}},
+    ),
     "relik-ie/relik-relation-extraction-small-wikipedia-ner": Relik.from_pretrained(
         "relik-ie/relik-relation-extraction-small-wikipedia-ner",
         device="cuda",
@@ -219,6 +234,16 @@ relik_models = {
             TaskType.TRIPLET: wikidata_retriever,
         },
     ),
+    "relik-ie/relik-relation-extraction-large": Relik.from_pretrained(
+        "relik-ie/relik-relation-extraction-large",
+        index={
+            TaskType.TRIPLET:wikidata_index,
+        },
+        device="cuda",
+        retriever={
+            TaskType.TRIPLET: wikidata_retriever,
+        },
+    ),
     "relik-ie/relik-entity-linking-large-robust": Relik.from_pretrained(
         "relik-ie/relik-entity-linking-large-robust",
         index={