abokbot commited on
Commit
7a3f7ed
·
1 Parent(s): d51ec0e

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
  @st.cache_resource
14
  def load_embedding():
15
  print("Loading embedding...")
16
- path = hf_hub_download(repo_id="abokbot/wikipedia-embedding", filename="simple_wikipedia_embedding.pt")
17
  wikipedia_embedding = torch.load(path, map_location=torch.device('cpu'))
18
  print("Embedding loaded!")
19
  return wikipedia_embedding
 
13
  @st.cache_resource
14
  def load_embedding():
15
  print("Loading embedding...")
16
+ path = hf_hub_download(repo_id="abokbot/wikipedia-embedding", filename="wikipedia_en_embedding.pt")
17
  wikipedia_embedding = torch.load(path, map_location=torch.device('cpu'))
18
  print("Embedding loaded!")
19
  return wikipedia_embedding