Update app.py

app.py

@@ -10,13 +10,17 @@ st.header("Wikipedia Search Engine app")
 st_model_load = st.text('Loading wikipedia embedding...')
 
 @st.cache_resource
-def load_model():
+def load_load_embedding():
     print("Loading embedding...")
     hf_hub_download(repo_id="abokbot/wikipedia-embedding", filename="simple_wikipedia_embedding.pt")
     wikipedia_embedding = torch.load("wikipedia-embedding/simple_wikipedia_embedding.pt") 
     print("Embedding loaded!")
     return wikipedia_embedding
 
+wikipedia_embedding = load_embedding()
+st.success('Embedding loaded!')
+st_model_load.text("")
+
 """