Hugging Face's logo

Spaces:
lefuuu
/
series_rec
Running

App Files Community
lefuuu commited on
Commit
bbae0f9
·
verified ·
1 Parent(s): e6857a5

Update pages/page_02.py

Browse files
Files changed (1) hide show
  1. pages/page_02.py +1 -1
pages/page_02.py CHANGED
@@ -15,7 +15,7 @@ def load_model():
15
 
16
  @st.cache_data
17
  def load_vectors():
18
- return torch.load("models/description_vectors_MiniLM-L12-v2.pt")
19
 
20
 
21
  def search_series(query, top_k=5):
 
15
 
16
  @st.cache_data
17
  def load_vectors():
18
+ return torch.load("models/description_vectors_MiniLM-L12-v2.pt", map_location=torch.device('cpu'))
19
 
20
 
21
  def search_series(query, top_k=5):