Hugging Face's logo

Spaces:
iNoBo
/
scinobo-fos-classification
Sleeping

App Files Community
sotiriskot commited on
Commit
ef43fb2
·
verified ·
1 Parent(s): 2d24c34

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_map="cuda"
11
  )
 
7
  model = SentenceTransformer(
8
  "all-mpnet-base-v2",
9
  cache_folder=CACHE_DIR,
10
+ device="cuda"
11
  )