abokbot commited on
Commit
84b4358
1 Parent(s): cd97c23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ st_model_load = st.text('Loading wikipedia embedding...')
13
  def load_embedding():
14
  print("Loading embedding...")
15
  path = hf_hub_download(repo_id="abokbot/wikipedia-embedding", filename="simple_wikipedia_embedding.pt")
16
- wikipedia_embedding = torch.load(path)
17
  print("Embedding loaded!")
18
  return wikipedia_embedding
19
 
 
13
  def load_embedding():
14
  print("Loading embedding...")
15
  path = hf_hub_download(repo_id="abokbot/wikipedia-embedding", filename="simple_wikipedia_embedding.pt")
16
+ wikipedia_embedding = torch.load(path, map_location=torch.device('cpu'))
17
  print("Embedding loaded!")
18
  return wikipedia_embedding
19