Update app.py

app.py

@@ -164,7 +164,7 @@ ner_type_retriever = GoldenSillyRetriever(
             )
 
 relik_available_models = [
-    "relik-ie/relik-reader-small-cie-wikipedia",
+    "relik-ie/relik-cie-small",
     "sapienzanlp/relik-entity-linking-large",
     "relik-ie/relik-entity-linking-large-robust",
     "relik-ie/relik-relation-extraction-small-wikipedia-ner",
@@ -179,9 +179,8 @@ relik_models = {
         retriever=wikipedia_retriever,
         reader_kwargs={"dataset_kwargs": {"use_nme": True}},
     ),
-    "relik-ie/relik-reader-small-cie-wikipedia": Relik.from_pretrained(
-        "relik-ie/relik-reader-small-cie-wikipedia",
-        # reader="/media/data/relik/models/PereLluis13/relik-reader-deberta-large-wikipedia-aida-full-interleave",
+    "relik-ie/relik-cie-small": Relik.from_pretrained(
+        "relik-ie/relik-cie-small",
         device="cuda",
         index={
             TaskType.SPAN: wikipedia_index,