Upload folder using huggingface_hub
Browse files- Test_RAG.py +2 -2
Test_RAG.py
CHANGED
|
@@ -615,13 +615,13 @@ def bot(history, temperature, top_p, top_k, repetition_penalty, hide_full_prompt
|
|
| 615 |
if do_rag:
|
| 616 |
# t1 = Thread(target=rag_chain.invoke, args=({"input": history[-1][0]},))
|
| 617 |
input_text = history[-1][0]
|
| 618 |
-
response = llm.invoke(
|
| 619 |
print(response)
|
| 620 |
else:
|
| 621 |
input_text = rag_prompt_template.format(input=history[-1][0], context="")
|
| 622 |
# t1 = Thread(target=llm.invoke, args=(input_text,))
|
| 623 |
# input_text = history[-1][0]
|
| 624 |
-
response = llm.invoke(
|
| 625 |
print(response)
|
| 626 |
# t1.start()
|
| 627 |
|
|
|
|
| 615 |
if do_rag:
|
| 616 |
# t1 = Thread(target=rag_chain.invoke, args=({"input": history[-1][0]},))
|
| 617 |
input_text = history[-1][0]
|
| 618 |
+
response = llm.invoke(input_text)
|
| 619 |
print(response)
|
| 620 |
else:
|
| 621 |
input_text = rag_prompt_template.format(input=history[-1][0], context="")
|
| 622 |
# t1 = Thread(target=llm.invoke, args=(input_text,))
|
| 623 |
# input_text = history[-1][0]
|
| 624 |
+
response = llm.invoke(input_text)
|
| 625 |
print(response)
|
| 626 |
# t1.start()
|
| 627 |
|