Hugging Face's logo

Spaces:
iNoBo
/
scinobo-fos-classification
Running

App Files Community
sotiriskot commited on
Commit
218e814
•
1 Parent(s): 2ecadf5

Update download_model.py

Browse files
Files changed (1) hide show
  1. download_model.py +1 -1
download_model.py CHANGED
@@ -7,5 +7,5 @@ CACHE_DIR = "/app/src/fos/data"
7
  model = SentenceTransformer(
8
  "all-mpnet-base-v2",
9
  cache_folder=CACHE_DIR,
10
- device="cuda"
11
  )
 
7
  model = SentenceTransformer(
8
  "all-mpnet-base-v2",
9
  cache_folder=CACHE_DIR,
10
+ device="cpu"
11
  )