ishaan-mital commited on
Commit
c8e6347
1 Parent(s): 32e7aaa

requirement.txt

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -3,21 +3,21 @@ import os
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 = SentenceTransformer('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(
 
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(