Update retriever.py

retriever.py

@@ -6,7 +6,11 @@ from config import EMBEDDING_MODEL, TOP_K
 
 class Retriever:
     def __init__(self):
-        self.model = SentenceTransformer(EMBEDDING_MODEL)
+        self.model = SentenceTransformer(
+                    EMBEDDING_MODEL,
+                    device="cpu"
+                )
+
         self.index = None
         self.chunks = []