dhruv107 commited on
Commit
745c9ac
1 Parent(s): c7af238
Files changed (1) hide show
  1. pretrained.py +1 -1
pretrained.py CHANGED
@@ -64,7 +64,7 @@ if __name__ == '__main__':
64
  print(f'loading model...')
65
 
66
  # model = SentenceTransformer("all-MiniLM-L6-v2", cache_folder='./')
67
- model = SentenceTransformer("all-MiniLM-L6-v2", cache_folder='./')
68
  #model = SentenceTransformer("models--sentence-transformers--all-MiniLM-L6-v2/snapshots/1a310852cf8e58d22c5ebff537711d504ad4ad66")
69
  model.max_seq_length = 512
70
  print(f'model max lenght is :{model.max_seq_length}')
 
64
  print(f'loading model...')
65
 
66
  # model = SentenceTransformer("all-MiniLM-L6-v2", cache_folder='./')
67
+ model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2", cache_folder='./')
68
  #model = SentenceTransformer("models--sentence-transformers--all-MiniLM-L6-v2/snapshots/1a310852cf8e58d22c5ebff537711d504ad4ad66")
69
  model.max_seq_length = 512
70
  print(f'model max lenght is :{model.max_seq_length}')