Update app.py

app.py

@@ -5,14 +5,13 @@ import spacy
 from spacy.tokens import Doc, Span
 from relik import Relik
 from relik.inference.data.objects import TaskType, RelikOutput
-from relik.retriever.pytorch_modules import GoldenRetriever
 from relik.retriever.indexers.inmemory import InMemoryDocumentIndex
 from pyvis.network import Network
 
 # RELIK Models Setup
-wikipedia_retriever = GoldenRetriever("relik-ie/encoder-e5-base-v2-wikipedia", device="cuda")
+wikipedia_retriever = Relik.from_pretrained("relik-ie/encoder-e5-base-v2-wikipedia", device="cuda")
 wikipedia_index = InMemoryDocumentIndex.from_pretrained("relik-ie/encoder-e5-base-v2-wikipedia-index", index_precision="bf16", device="cuda")
-wikidata_retriever = GoldenRetriever("relik-ie/encoder-e5-small-v2-wikipedia-relations", device="cuda")
+wikidata_retriever = Relik.from_pretrained("relik-ie/encoder-e5-small-v2-wikipedia-relations", device="cuda")
 wikidata_index = InMemoryDocumentIndex.from_pretrained("relik-ie/encoder-e5-small-v2-wikipedia-relations-index", index_precision="bf16", device="cuda")
 
 relik_models = {