Update script.py

script.py

@@ -161,7 +161,7 @@ def generate_embeddings(metadata_file_path, root_dir):
 
     loader = DataLoader(test_dataset, batch_size=3, shuffle=False)
 
-    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    device = torch.device('cpu')
     model = timm.create_model(
         "timm/vit_large_patch14_reg4_dinov2.lvd142m", pretrained=False
     )
@@ -225,7 +225,7 @@ class FungiMEEModel(nn.Module):
         super().__init__()
 
         print("Setting up Pytorch Model")
-        self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+        self.device = torch.device('cpu')
         print(f"Using devide: {self.device}")
 
         self.date_embedding = MlpHead(
@@ -279,7 +279,7 @@ class FungiEnsembleModel(nn.Module):
         super().__init__()
 
         self.models = nn.ModuleList()
-        self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+        self.device = torch.device('cpu')
 
         for model in models:
             model = model.to(self.device)