Namitg02 commited on
Commit
2a5ffb8
·
verified ·
1 Parent(s): a4e4c09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from datasets import load_dataset
2
- dataset = load_dataset("Namitg02/Test")
3
  print(dataset)
4
 
5
  from langchain.docstore.document import Document as LangchainDocument
@@ -11,7 +11,7 @@ from sentence_transformers import SentenceTransformer
11
  from langchain_community.embeddings import HuggingFaceEmbeddings
12
  embedding_model = HuggingFaceEmbeddings(model_name = "all-MiniLM-L6-v2")
13
 
14
- from langchain.vectorstores import FAISS
15
  data = FAISS.from_documents(docs, embedding_model)
16
 
17
  #data = dataset["train"]
 
1
  from datasets import load_dataset
2
+ dataset = load_dataset("Namitg02/Test", streaming = False)
3
  print(dataset)
4
 
5
  from langchain.docstore.document import Document as LangchainDocument
 
11
  from langchain_community.embeddings import HuggingFaceEmbeddings
12
  embedding_model = HuggingFaceEmbeddings(model_name = "all-MiniLM-L6-v2")
13
 
14
+ from langchain_community.vectorstores import FAISS
15
  data = FAISS.from_documents(docs, embedding_model)
16
 
17
  #data = dataset["train"]