Spaces:
Sleeping
Sleeping
Commit
·
c8e6347
1
Parent(s):
32e7aaa
requirement.txt
Browse files
app.py
CHANGED
@@ -3,21 +3,21 @@ import os
|
|
3 |
import pinecone
|
4 |
import time
|
5 |
# from torch import cuda
|
6 |
-
|
7 |
# import PyPDF2
|
8 |
# import re
|
9 |
from langchain.vectorstores import Pinecone
|
10 |
from sentence_transformers import SentenceTransformer
|
11 |
|
12 |
|
13 |
-
|
14 |
# device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
|
15 |
|
16 |
-
|
17 |
-
|
18 |
-
#
|
19 |
-
#
|
20 |
-
|
21 |
|
22 |
# get API key from app.pinecone.io and environment from console
|
23 |
pinecone.init(
|
|
|
3 |
import pinecone
|
4 |
import time
|
5 |
# from torch import cuda
|
6 |
+
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|
7 |
# import PyPDF2
|
8 |
# import re
|
9 |
from langchain.vectorstores import Pinecone
|
10 |
from sentence_transformers import SentenceTransformer
|
11 |
|
12 |
|
13 |
+
embed_model_id = 'sentence-transformers/all-MiniLM-L6-v2'
|
14 |
# device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
|
15 |
|
16 |
+
embed_model = HuggingFaceEmbeddings(
|
17 |
+
model_name=embed_model_id,
|
18 |
+
# model_kwargs={'device': device},
|
19 |
+
# encode_kwargs={'device': device, 'batch_size': 32}
|
20 |
+
)
|
21 |
|
22 |
# get API key from app.pinecone.io and environment from console
|
23 |
pinecone.init(
|