MBilal-72 commited on
Commit
f5b1b84
·
verified ·
1 Parent(s): ebe973b

model_kwargs={"device": "cpu"} # <-- Add this line

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -48,6 +48,7 @@ class GEOSEOApp:
48
  self.embeddings = HuggingFaceEmbeddings(
49
  model_name="sentence-transformers/all-MiniLM-L6-v2",
50
  cache_folder="./hf_cache",
 
51
  )
52
 
53
  def setup_parsers(self):
 
48
  self.embeddings = HuggingFaceEmbeddings(
49
  model_name="sentence-transformers/all-MiniLM-L6-v2",
50
  cache_folder="./hf_cache",
51
+ model_kwargs={"device": "cpu"} # <-- Add this line
52
  )
53
 
54
  def setup_parsers(self):