Update app.py

app.py

@@ -46,7 +46,8 @@ class GradioWindow():
         self.model_type = SAM_ENCODER_VERSION
         self.SAM_CHECKPOINT_PATH = SAM_CHECKPOINT_PATH
 
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        # self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.device = "cpu"
         # for debug
         # self.augmenter = None
         self.augmenter = Augmenter(device=self.device)
@@ -176,7 +177,7 @@ class GradioWindow():
             device=self.device
             )
         
-        print("MODELS LOADED!")
+        print("MODELS LOADED! Device:", self.device)
 
     def change_mask_type(self, image, is_segmmask):
         self.selected_mask = None