Jerome2046 commited on
Commit
dcd0e32
1 Parent(s): b55fd41

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -8,6 +8,8 @@ pipeline_tag: text-classification
8
 
9
  加载代码:
10
  ```python
 
 
11
  import torch
12
  import os
13
  from transformers import AutoConfig, AutoModel, AutoTokenizer
@@ -23,9 +25,8 @@ for k, v in prefix_state_dict.items():
23
  if k.startswith("transformer.prefix_encoder."):
24
  new_prefix_state_dict[k[len("transformer.prefix_encoder."):]] = v
25
  model.transformer.prefix_encoder.load_state_dict(new_prefix_state_dict)
26
-
27
  model.half().cuda().eval()
28
 
29
- response, history = model.chat(tokenizer, "说出下列句子的情绪类型:到北京五环25分钟车程,938离我住处只有300米,饮食一条街更是北京饮食街的翻版", history=[])
30
  print(response)
31
  ```
 
8
 
9
  加载代码:
10
  ```python
11
+ !git clone https://huggingface.co/Jerome2046/glm-emotion
12
+
13
  import torch
14
  import os
15
  from transformers import AutoConfig, AutoModel, AutoTokenizer
 
25
  if k.startswith("transformer.prefix_encoder."):
26
  new_prefix_state_dict[k[len("transformer.prefix_encoder."):]] = v
27
  model.transformer.prefix_encoder.load_state_dict(new_prefix_state_dict)
 
28
  model.half().cuda().eval()
29
 
30
+ response, history = model.chat(tokenizer, "说出'到北京五环25分钟车程,938离我住处只有300米,饮食一条街更是北京饮食街的翻版'此文段表达的情绪类型", history=[])
31
  print(response)
32
  ```