abokbot commited on
Commit
cd97c23
1 Parent(s): d45163a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ st_model_load = st.text('Loading wikipedia embedding...')
12
  @st.cache_resource
13
  def load_embedding():
14
  print("Loading embedding...")
15
- hf_hub_download(repo_id="abokbot/wikipedia-embedding", filename="simple_wikipedia_embedding.pt")
16
- wikipedia_embedding = torch.load("simple_wikipedia_embedding.pt")
17
  print("Embedding loaded!")
18
  return wikipedia_embedding
19
 
 
12
  @st.cache_resource
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