crodri commited on
Commit
6291eed
·
verified ·
1 Parent(s): 944ef9e

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -115,7 +115,7 @@ class RAG:
115
  chat_completion = client.chat.completions.create(
116
  model="tgi",
117
  messages=[
118
- {"role": "user", "content": instruction}
119
  ],
120
  temperature=model_parameters["temperature"],
121
  max_tokens=model_parameters["max_new_tokens"],
 
115
  chat_completion = client.chat.completions.create(
116
  model="tgi",
117
  messages=[
118
+ {"role": "user", "content": query}
119
  ],
120
  temperature=model_parameters["temperature"],
121
  max_tokens=model_parameters["max_new_tokens"],