Tensor to cpu

app.py

@@ -60,7 +60,7 @@ def bi_encode(bi_encoder_name,passages, dataset_name='scifact-pl'):
             name = 'mcontriever'
 
         corpus_embeddings_name = "-".join([name, dataset_name, "corpus"])
-        corpus_embeddings = torch.load(corpus_embeddings_name)
+        corpus_embeddings = torch.load(corpus_embeddings_name, map_location=torch.device('cpu'))
 
 
     st.success(f"Embeddings computed. Shape: {corpus_embeddings.shape}")