Update README.md

README.md

@@ -52,8 +52,8 @@ import torch
 import torch.nn.functional as F
 from transformers import BertForSequenceClassification, BertTokenizer
 
-model = BertForSequenceClassification.from_pretrained('lhallee/SYNTERACT') # load model
-tokenizer = BertTokenizer.from_pretrained('lhallee/SYNTERACT') # load tokenizer
+model = BertForSequenceClassification.from_pretrained('GleghornLab/SYNTERACT') # load model
+tokenizer = BertTokenizer.from_pretrained('GleghornLab/SYNTERACT') # load tokenizer
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu') # gather device
 model.to(device) # move to device
 model.eval() # put in eval mode