AdrienB134 commited on
Commit
9b37798
·
verified ·
1 Parent(s): 3a481e5

Update rag_demo/rag/retriever.py

Browse files
Files changed (1) hide show
  1. rag_demo/rag/retriever.py +1 -1
rag_demo/rag/retriever.py CHANGED
@@ -112,7 +112,7 @@ class RAGPipeline:
112
  context += "\n Document: "
113
  context += chunk.content
114
  api = InferenceClient(
115
- model="Qwen/Qwen2.5-72B-Instruct",
116
  token=os.getenv("HF_API_TOKEN"),
117
  )
118
  answer_generation_template = AnswerGenerationTemplate()
 
112
  context += "\n Document: "
113
  context += chunk.content
114
  api = InferenceClient(
115
+ model="meta-llama/Llama-3.3-70B-Instruct",
116
  token=os.getenv("HF_API_TOKEN"),
117
  )
118
  answer_generation_template = AnswerGenerationTemplate()