Carlosito16 commited on
Commit
ea780f0
1 Parent(s): ac71947

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -43,9 +43,8 @@ st.markdown(f"Number of chunked texts: {len(chunked_text)}")
43
 
44
  embedding_model = HuggingFaceInstructEmbeddings(model_name='hkunlp/instructor-base',
45
  model_kwargs = {'device': torch.device('cuda' if torch.cuda.is_available() else 'cpu')})
46
- persist_directory = 'db_chunk_500'
47
  db_chunk_500 = Chroma.from_documents(documents= chunked_text,
48
- embedding= embedding_model,
49
- persist_directory=persist_directory)
50
 
51
  print("load done")
 
43
 
44
  embedding_model = HuggingFaceInstructEmbeddings(model_name='hkunlp/instructor-base',
45
  model_kwargs = {'device': torch.device('cuda' if torch.cuda.is_available() else 'cpu')})
46
+
47
  db_chunk_500 = Chroma.from_documents(documents= chunked_text,
48
+ embedding= embedding_model)
 
49
 
50
  print("load done")