Hugging Face's logo

Spaces:
ki-adventskalender
/
ki-adventskalender-aufgabe-24
Sleeping

App Files
ki-adventskalender commited on
Commit
185459e
·
verified ·
1 Parent(s): aa2f1bd

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import sys
11
 
12
  # Load an embedding model
13
  model = SentenceTransformer('mixedbread-ai/deepset-mxbai-embed-de-large-v1')
14
- ideal_embedding = torch.load("ideal_embedding.pt", weights_only=True)
15
  keywords = json.loads(os.environ.get('ENV_LIST_EXAMPLE'))
16
  region_keywords = json.loads(os.environ.get('ENV_DICT_EXAMPLE'))
17
 
 
11
 
12
  # Load an embedding model
13
  model = SentenceTransformer('mixedbread-ai/deepset-mxbai-embed-de-large-v1')
14
+ ideal_embedding = torch.load("ideal_embedding.pt", weights_only=True, map_location=torch.device("cpu"))
15
  keywords = json.loads(os.environ.get('ENV_LIST_EXAMPLE'))
16
  region_keywords = json.loads(os.environ.get('ENV_DICT_EXAMPLE'))
17