ryanrwatkins commited on
Commit
5a259af
·
verified ·
1 Parent(s): 145a5a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -98,7 +98,7 @@ actor_description = {"All Needs Experts": "<div style='float: left;margin: 0px 5
98
 
99
  from huggingface_hub import InferenceApi
100
 
101
- api = InferenceApi(repo_id="sentence-transformers/all-MiniLM-l6-v2", token=HF_key)
102
  response = api(inputs="test sentence")
103
 
104
  print("API response:", response)
@@ -237,7 +237,7 @@ def select_embeddings_model(LLM_service="HuggingFace"):
237
  embeddings = HuggingFaceInferenceAPIEmbeddings(
238
  api_key=HF_key,
239
  #model_name="thenlper/gte-large"
240
- model_name="sentence-transformers/all-MiniLM-l6-v2"
241
  )
242
  print(embeddings)
243
  return embeddings
 
98
 
99
  from huggingface_hub import InferenceApi
100
 
101
+ api = InferenceApi(repo_id="sentence-transformers/all-MiniLM-L6-v2", token=HF_key)
102
  response = api(inputs="test sentence")
103
 
104
  print("API response:", response)
 
237
  embeddings = HuggingFaceInferenceAPIEmbeddings(
238
  api_key=HF_key,
239
  #model_name="thenlper/gte-large"
240
+ model_name="sentence-transformers/all-MiniLM-L6-v2"
241
  )
242
  print(embeddings)
243
  return embeddings