achuthc1298 commited on
Commit
cd3b67b
·
verified ·
1 Parent(s): 52fb95e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -19,8 +19,10 @@ import io
19
  def create_doc_tools(document_fp: str, doc_name: str, verbose: bool = True) -> Tuple[QueryEngineTool,]:
20
  documents = SimpleDirectoryReader(input_files=[document_fp]).load_data()
21
 
22
- Settings.llm = Groq(model="mixtral-8x7b-32768")
23
- Settings.embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-mpnet-base-v2")
 
 
24
 
25
  load_dir_path = f"/home/user/app/agentic_index_st/{doc_name}"
26
  storage_context = StorageContext.from_defaults(persist_dir=load_dir_path)
 
19
  def create_doc_tools(document_fp: str, doc_name: str, verbose: bool = True) -> Tuple[QueryEngineTool,]:
20
  documents = SimpleDirectoryReader(input_files=[document_fp]).load_data()
21
 
22
+ llm = Groq(model="mixtral-8x7b-32768")
23
+ embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-mpnet-base-v2")
24
+ Settings.llm = llm
25
+ Settings.embed_model = embed_model
26
 
27
  load_dir_path = f"/home/user/app/agentic_index_st/{doc_name}"
28
  storage_context = StorageContext.from_defaults(persist_dir=load_dir_path)