niranjan9795 commited on
Commit
e620b8b
·
verified ·
1 Parent(s): 46b544f

Update GPU to CPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def split_pages(pages):
40
  def create_embedding():
41
  embeddings = HuggingFaceEmbeddings(
42
  model_name="sentence-transformers/all-MiniLM-L6-v2",
43
- model_kwargs={'device': 'cuda'}
44
  )
45
  return embeddings
46
 
 
40
  def create_embedding():
41
  embeddings = HuggingFaceEmbeddings(
42
  model_name="sentence-transformers/all-MiniLM-L6-v2",
43
+ model_kwargs={'device': 'cpu'}
44
  )
45
  return embeddings
46