Hugging Face's logo

Spaces:
MSHS-Neurosurgery-Research
/
NSQIP-ACC
Sleeping

App Files Community
mertkarabacak commited on
Commit
6b815f3
·
verified ·
1 Parent(s): 61edbed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ y1_valid = x1_valid.pop('OUTCOME')
46
 
47
  #Training model.
48
  from tabpfn import TabPFNClassifier
49
- tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=8)
50
 
51
  y1_model = tabpfn
52
  y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
 
46
 
47
  #Training model.
48
  from tabpfn import TabPFNClassifier
49
+ tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=8)
50
 
51
  y1_model = tabpfn
52
  y1_model = y1_model.fit(x1, y1, overwrite_warning=True)