Upload folder using huggingface_hub

Test_RAG.py

@@ -613,17 +613,22 @@ def bot(history, temperature, top_p, top_k, repetition_penalty, hide_full_prompt
         llm.pipeline._forward_params["stopping_criteria"] = StoppingCriteriaList(stop_tokens)
 
     if do_rag:
-        t1 = Thread(target=rag_chain.invoke, args=({"input": history[-1][0]},))
+        # t1 = Thread(target=rag_chain.invoke, args=({"input": history[-1][0]},))
+        input_text = history[-1][0]
+        response = rag_chain.run(input_text)
     else:
-        input_text = rag_prompt_template.format(input=history[-1][0], context="")
-        t1 = Thread(target=llm.invoke, args=(input_text,))
-    t1.start()
+        # input_text = rag_prompt_template.format(input=history[-1][0], context="")
+        # t1 = Thread(target=llm.invoke, args=(input_text,))
+        input_text = history[-1][0]
+        response = rag_chain.run(input_text)
+    # t1.start()
 
     # Initialize an empty string to store the generated text
-    partial_text = ""
-    for new_text in streamer:
-        partial_text = text_processor(partial_text, new_text)
-        history[-1][1] = partial_text
+    # partial_text = ""
+    # for new_text in streamer:
+    #     partial_text = text_processor(partial_text, new_text)
+    #     history[-1][1] = partial_text
+        history[-1][1] = response
         yield history