JiakaiDu commited on
Commit
cbcb9b9
1 Parent(s): 6ccaa71

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. Test_RAG.py +2 -0
Test_RAG.py CHANGED
@@ -616,11 +616,13 @@ def bot(history, temperature, top_p, top_k, repetition_penalty, hide_full_prompt
616
  # t1 = Thread(target=rag_chain.invoke, args=({"input": history[-1][0]},))
617
  input_text = history[-1][0]
618
  response = rag_chain.run(input_text)
 
619
  else:
620
  # input_text = rag_prompt_template.format(input=history[-1][0], context="")
621
  # t1 = Thread(target=llm.invoke, args=(input_text,))
622
  input_text = history[-1][0]
623
  response = rag_chain.run(input_text)
 
624
  # t1.start()
625
 
626
  # Initialize an empty string to store the generated text
 
616
  # t1 = Thread(target=rag_chain.invoke, args=({"input": history[-1][0]},))
617
  input_text = history[-1][0]
618
  response = rag_chain.run(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 = rag_chain.run(input_text)
625
+ print(response)
626
  # t1.start()
627
 
628
  # Initialize an empty string to store the generated text