sjiang1 commited on
Commit
aab1a7d
·
1 Parent(s): 50bab77

Add "feature extraction" as our pipeline

Browse files
Files changed (1) hide show
  1. README.md +5 -4
README.md CHANGED
@@ -1,12 +1,13 @@
1
  ---
2
- language:
3
- - multilingual
4
  tags:
5
  - code
6
  - sentence embedding
7
- license: "mit"
8
  datasets:
9
  - CodeSearchNet
 
10
  ---
11
 
12
  # Model Card for CodeCSE
@@ -36,4 +37,4 @@ batch = prepare_inputs(nl_json, tokenizer, args)
36
  nl_inputs = batch[3]
37
  with torch.no_grad():
38
  nl_vec = model(input_ids=nl_inputs, sent_emb="nl")[1]
39
- ```
 
1
  ---
2
+ language:
3
+ - multilingual
4
  tags:
5
  - code
6
  - sentence embedding
7
+ license: mit
8
  datasets:
9
  - CodeSearchNet
10
+ pipeline_tag: feature-extraction
11
  ---
12
 
13
  # Model Card for CodeCSE
 
37
  nl_inputs = batch[3]
38
  with torch.no_grad():
39
  nl_vec = model(input_ids=nl_inputs, sent_emb="nl")[1]
40
+ ```