Update app.py

app.py

@@ -26,10 +26,10 @@ with col3:
 
 if model_type == 'rubert-tiny2':
     model = SentenceTransformer('cointegrated/rubert-tiny2')
-    emb = torch.load('resources/corpus_embeddings_rub.pth')
+    emb = torch.load('resources/corpus_embeddings_rub.pth', map_location=torch.device('cpu'))
 else:
     model = SentenceTransformer('msmarco-MiniLM-L-12-v3')
-    emb = torch.load('resources/corpus_embeddings_ms.pth')
+    emb = torch.load('resources/corpus_embeddings_ms.pth', map_location=torch.device('cpu'))
 
 range_years = st.slider("В каком году вышел фильм?", min_value=df['year'].unique().min(), 
                          max_value=df['year'].unique().max(),