Spaces:

Sakil
/

CSVQConnect

Sleeping

Sakil commited on Dec 1, 2023

Commit

99e5370

1 Parent(s): cfa7d92

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,8 +30,7 @@ if uploaded_file :
                 'delimiter': ','})
     data = loader.load()
     #st.json(data)
-    embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2',
-                                       model_kwargs={'device': 'cpu'})
     db = FAISS.from_documents(data, embeddings)
     db.save_local(DB_FAISS_PATH)

                 'delimiter': ','})
     data = loader.load()
     #st.json(data)
+    embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2')
     db = FAISS.from_documents(data, embeddings)
     db.save_local(DB_FAISS_PATH)