araeyn commited on
Commit
e9884ae
1 Parent(s): 27bc640

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -40,15 +40,11 @@ persist_directory = 'db'
40
  model_name = "BAAI/bge-large-en"
41
  model_kwargs = {'device': 'cpu'}
42
  encode_kwargs = {'normalize_embeddings': True}
43
- """embedding = HuggingFaceBgeEmbeddings(
44
  model_name=model_name,
45
  model_kwargs=model_kwargs,
46
  encode_kwargs=encode_kwargs,
47
  show_progress=True,
48
- )"""
49
- embedding = HuggingFaceInferenceAPIEmbeddings(
50
- api_key=os.environ["HUGGINGFACE_API_KEY"],
51
- model_name="sentence-transformers/all-MiniLM-l6-v2"
52
  )
53
 
54
 
 
40
  model_name = "BAAI/bge-large-en"
41
  model_kwargs = {'device': 'cpu'}
42
  encode_kwargs = {'normalize_embeddings': True}
43
+ embedding = HuggingFaceBgeEmbeddings(
44
  model_name=model_name,
45
  model_kwargs=model_kwargs,
46
  encode_kwargs=encode_kwargs,
47
  show_progress=True,
 
 
 
 
48
  )
49
 
50