Update appStore/rag.py
Browse files- appStore/rag.py +1 -1
appStore/rag.py
CHANGED
@@ -44,7 +44,7 @@ def run_query(context, label, model_sel_name):
|
|
44 |
client = InferenceClient(model_sel_name, token = hf_token)
|
45 |
|
46 |
# instantiate ChatCompletion as a generator object (stream is set to True)
|
47 |
-
chat_completion = client.
|
48 |
messages=messages,
|
49 |
stream=True
|
50 |
)
|
|
|
44 |
client = InferenceClient(model_sel_name, token = hf_token)
|
45 |
|
46 |
# instantiate ChatCompletion as a generator object (stream is set to True)
|
47 |
+
chat_completion = client.chat_completion(
|
48 |
messages=messages,
|
49 |
stream=True
|
50 |
)
|