Spaces:
Runtime error
Runtime error
Alexandre
commited on
Update rag.py
Browse files
rag.py
CHANGED
|
@@ -113,7 +113,7 @@ class RAG:
|
|
| 113 |
chat_completion = client.chat.completions.create(
|
| 114 |
model="tgi",
|
| 115 |
messages=[
|
| 116 |
-
{"role": "user", "content":
|
| 117 |
],
|
| 118 |
temperature=model_parameters["temperature"],
|
| 119 |
max_tokens=model_parameters["max_new_tokens"],
|
|
|
|
| 113 |
chat_completion = client.chat.completions.create(
|
| 114 |
model="tgi",
|
| 115 |
messages=[
|
| 116 |
+
{"role": "user", "content": query}
|
| 117 |
],
|
| 118 |
temperature=model_parameters["temperature"],
|
| 119 |
max_tokens=model_parameters["max_new_tokens"],
|