Update README.md

README.md

@@ -42,8 +42,9 @@ device = torch.device("cuda:1" if torch.cuda.is_available() else "cpu")
 device = 'cpu'
 
 # load model
-clip_tokenizer = AutoTokenizer.from_pretrained("./segvol")
-model = AutoModel.from_pretrained("./segvol", trust_remote_code=True, test_mode=True)
+# 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"
+clip_tokenizer = AutoTokenizer.from_pretrained("yuxindu/segvol")
+model = AutoModel.from_pretrained("yuxindu/segvol", trust_remote_code=True, test_mode=True)
 model.model.text_encoder.tokenizer = clip_tokenizer
 model.eval()
 model.to(device)