Update README.md

README.md

@@ -38,7 +38,7 @@ from transformers import AutoModel, AutoTokenizer
 import torch
 
 # get device
-device = torch.device("cuda:1" if torch.cuda.is_available() else "cpu")
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 # load model
 # IF you cannot connect to huggingface.co, you can download the repo and set from_pretrained path as the loacl dir path, replacing "yuxindu/segvol"
@@ -114,7 +114,7 @@ from transformers import AutoModel, AutoTokenizer
 import torch
 
 # get device
-device = torch.device("cuda:1" if torch.cuda.is_available() else "CPU")
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 # load model
 # IF you cannot connect to huggingface.co, you can download the repo and set from_pretrained path as the loacl dir path, replacing "yuxindu/segvol"