Spaces:

TabPFN
/

TabPFNPrediction

Runtime error

TabPFN commited on Mar 9, 2023

Commit

def7fc3

1 Parent(s): a0b5459

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,9 @@ import os
 import matplotlib.pyplot as plt
 from matplotlib.colors import ListedColormap
-classifier = TabPFNClassifier(base_path=tabpfn_path, device='cpu', N_ensemble_configurations=4)
 def compute(df_table):

 import matplotlib.pyplot as plt
 from matplotlib.colors import ListedColormap
+default_device = "cuda:0" if torch.cuda.is_available() else "cpu:0"
+classifier = TabPFNClassifier(base_path=tabpfn_path, device=default_device, N_ensemble_configurations=4)
 def compute(df_table):