Views
No views yet
pip install ragatouille==0.0.8 transformers==4.491from ragatouille import RAGPretrainedModel
2from ragatouille.utils import get_wikipedia_page
3
4# load the model
5model = RAGPretrainedModel.from_pretrained("C4IR-RW/kiny-colbert-free")
6
7# create and index knowledge base
8# the first article is about maiz farming, the 2nd about cattle vaccination
9knowledge_base = [
10 "Ikigori ni igihingwa cy’ingenzi gikura neza mubutaka bukungahaye kandi budatunganyije amazi, ahari izuba rihagije. Abahinzi bakwiye gutera imbuto mu ntiriranywa y’ibihe by’imvura kugira ngo ibihingwa bikiri bito bibone amazi ahagije. Kubagara kenshi no gushyiramo ifumbire y'imborera cyangwa iy'imvugururaburinganire ni ingenzi kugira ngo umusaruro ube mwiza. Iyo amashushu abaye ay'ikigina n'impeke zikaba zishitse, ibigori biba bishobora gusarurwa no guhunikwa.",
11 "Gukingira inka ni intambwe y'ingenzi mu kwirinda indwara zica nka Borigo, Uburenge, n'Igifuruta. Kuriha inkingo hakiri kare birahendutse kuruta kuvura amatungo yarwaye. Abahinzi bakwiye gukorana bya hafi n'umuvuzi w'amatungo kugira ngo bagire ingengabihe y'inkingo ihoraho, bityo basigasire ubuzima n'umusaruro w'amatungo yabo."
12]
13
14index_path = model.index(index_name="my_index", collection=knowledge_base)
15
16# retrieve data
17
18# question about maize farming
19question = "Mfite ikibazo kijyanye n'ubuhinzi bw'ibigori"
20
21RAG = RAGPretrainedModel.from_index(index_path)
22results = RAG.search(question)
23
24print(results)
25
26"""
27Outputs:
28
29[{'content': "Ikigori ni igihingwa cy’ingenzi gikura neza mubutaka bukungahaye kandi budatunganyije amazi, ahari izuba rihagije. Abahinzi bakwiye gutera imbuto mu ntiriranywa y’ibihe by’imvura kugira ngo ibihingwa bikiri bito bibone amazi ahagije. Kubagara kenshi no gushyiramo ifumbire y'imborera cyangwa iy'imvugururaburinganire ni ingenzi kugira ngo umusaruro ube mwiza. Iyo amashushu abaye ay'ikigina n'impeke zikaba zishitse, ibigori biba bishobora gusarurwa no guhunikwa.",
30 'score': 19.476581573486328,
31 'rank': 1,
32 'document_id': '74d59927-b84f-4140-a35a-17477787453b',
33 'passage_id': 0}]
34"""