MBilal-72 commited on
Commit
f61e5a0
·
verified ·
1 Parent(s): cc4861a

Update app.py

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