Spaces:

oucgc1996
/

pyuul_kmeans

Sleeping

oucgc1996 commited on Aug 2, 2023

Commit

a6883fb

1 Parent(s): e848314

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ def setup_seed(seed):
     random.seed(seed)
     torch.backends.cudnn.deterministic = True
 setup_seed(100)
-device = "cuda"
 def add_file_to_folder(new_file_path, folder):
     ls = os.listdir(new_file_path)

     random.seed(seed)
     torch.backends.cudnn.deterministic = True
 setup_seed(100)
+device = "cpu"
 def add_file_to_folder(new_file_path, folder):
     ls = os.listdir(new_file_path)