ccm commited on
Commit
2096682
·
verified ·
1 Parent(s): 7757934

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ PUBLICATIONS_TO_RETRIEVE = 5
32
 
33
  def embedding(
34
  model_name: str = "all-MiniLM-L12-v2",
35
- device: str = "mps",
36
  normalize_embeddings: bool = False,
37
  ) -> langchain_huggingface.HuggingFaceEmbeddings:
38
  """
 
32
 
33
  def embedding(
34
  model_name: str = "all-MiniLM-L12-v2",
35
+ device: str = "cuda",
36
  normalize_embeddings: bool = False,
37
  ) -> langchain_huggingface.HuggingFaceEmbeddings:
38
  """