Spaces:
Running
Running
Update app.py
Browse files
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="
|
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
|