jessicayjm commited on
Commit
f798a31
1 Parent(s): 496cffd

update sample code

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -50,7 +50,7 @@ class SNN(nn.Module):
50
  pred = self.cos(meanPooled1, meanPooled2)
51
  return pred
52
 
53
- checkpoint_path = 'upload_version/empathy-appraisal-alignment.pt'
54
 
55
  tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
56
  model = SNN('sentence-transformers/all-mpnet-base-v2').to('cuda')
 
50
  pred = self.cos(meanPooled1, meanPooled2)
51
  return pred
52
 
53
+ checkpoint_path = 'your_path_to/empathy-appraisal-alignment.pt'
54
 
55
  tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
56
  model = SNN('sentence-transformers/all-mpnet-base-v2').to('cuda')