Spaces:

vishwask
/

rag

Sleeping

vishwask commited on Feb 28, 2024

Commit

4b1fab8

verified ·

1 Parent(s): aec6363

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -141,7 +141,7 @@ def initialize_database(list_file_obj, chunk_size, chunk_overlap, progress=gr.Pr
 def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, progress=gr.Progress()):
     # print("llm_option",llm_option)
-    llm_name = list_llm[llm_option]
     print("llm_name: ",llm_name)
     qa_chain = initialize_llmchain(llm_name, llm_temperature, max_tokens, top_k, vector_db, progress)
     return qa_chain, "Complete!"

 def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, progress=gr.Progress()):
     # print("llm_option",llm_option)
+    llm_name = llm_model
     print("llm_name: ",llm_name)
     qa_chain = initialize_llmchain(llm_name, llm_temperature, max_tokens, top_k, vector_db, progress)
     return qa_chain, "Complete!"