owaiskha9654 commited on
Commit
d578004
·
verified ·
1 Parent(s): 8194e39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -29,11 +29,11 @@ for fname in file_paths:
29
 
30
 
31
  embedding_model = HuggingFaceInstructEmbeddings(
32
- #model_name="hkunlp/instructor-large",
33
  #model_name="jinaai/jina-embedding-b-en-v1",
34
- model_name="WhereIsAI/UAE-Large-V1",
35
- model_kwargs={"device": "cuda"}
36
- #model_kwargs={"device": "cpu"}
37
  )
38
 
39
  df_documents_chunks = pd.DataFrame({"doc_pages":page_contents})
 
29
 
30
 
31
  embedding_model = HuggingFaceInstructEmbeddings(
32
+ model_name="hkunlp/instructor-base",
33
  #model_name="jinaai/jina-embedding-b-en-v1",
34
+ #model_name="WhereIsAI/UAE-Large-V1",
35
+ #model_kwargs={"device": "cuda"}
36
+ model_kwargs={"device": "cpu"}
37
  )
38
 
39
  df_documents_chunks = pd.DataFrame({"doc_pages":page_contents})