myentity commited on
Commit
5804ce1
·
verified ·
1 Parent(s): 3c4d5ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -72,7 +72,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
72
  tokenizer = AutoTokenizer.from_pretrained("AIPsy/bert-base-client-topic-classification-eng")
73
  model = AutoModelForSequenceClassification.from_pretrained("AIPsy/bert-base-client-topic-classification-eng")
74
 
75
- text = "You don't even seem to care that I'm mad at you."
76
 
77
  encoding = tokenizer(
78
  text,
@@ -83,7 +83,7 @@ encoding = tokenizer(
83
  output = model(encoding['input_ids'], encoding['attention_mask']).logits
84
  result = np.argmax(output.detach().numpy(), axis=-1)
85
  print(id2label[result[0]])
86
- 'Experience and Regulation of Anger'
87
  ```
88
 
89
  ## Dataset
 
72
  tokenizer = AutoTokenizer.from_pretrained("AIPsy/bert-base-client-topic-classification-eng")
73
  model = AutoModelForSequenceClassification.from_pretrained("AIPsy/bert-base-client-topic-classification-eng")
74
 
75
+ text = "That in essense it's right; it's responsibility."
76
 
77
  encoding = tokenizer(
78
  text,
 
83
  output = model(encoding['input_ids'], encoding['attention_mask']).logits
84
  result = np.argmax(output.detach().numpy(), axis=-1)
85
  print(id2label[result[0]])
86
+ 'Guilt and Responsibility'
87
  ```
88
 
89
  ## Dataset