araeyn commited on
Commit
b79a299
·
verified ·
1 Parent(s): 52fb498

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -65,7 +65,16 @@ print("-------")
65
 
66
  persist_directory = 'db'
67
 
68
- embedding = HuggingFaceEmbeddings()
 
 
 
 
 
 
 
 
 
69
 
70
  print()
71
  print("-------")
 
65
 
66
  persist_directory = 'db'
67
 
68
+ # embedding = HuggingFaceInferenceAPIEmbeddings(api_key=os.environ["HUGGINGFACE_API_KEY"], model=)
69
+ model_name = "BAAI/bge-large-en"
70
+ model_kwargs = {'device': 'cpu'}
71
+ encode_kwargs = {'normalize_embeddings': True}
72
+ hf = HuggingFaceBgeEmbeddings(
73
+ model_name=model_name,
74
+ model_kwargs=model_kwargs,
75
+ encode_kwargs=encode_kwargs,
76
+ show_progress=True,
77
+ )
78
 
79
  print()
80
  print("-------")