Call Contriever in a separate file

[lflage]

Main function was too convoluted so I decide to run Contriever in another file.

There should be a function with inputs:

• Query
• k (as in top-k documents)

Returning:

• k documents

Function must load Contriever, embed the query, calculate score, return top-k

https://github.com/lflage/HalluEval-RAG/blob/8aa2e02edab69ae488f2625d1e80a53ed9b72686/src/main.py#L28-L75

[lflage]

Question:

• Would it be faster to embed the queries beforehand and only load them?

Discussion:

• maybe not since I still need the query string for other models
• If I can load both simultaneously I might save some time