PavanDeepak commited on
Commit
a83329a
·
verified ·
1 Parent(s): 89883b6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -17
README.md CHANGED
@@ -33,27 +33,27 @@ The model uses the GELU activation function in its hidden layers and applies dro
33
 
34
  ## Example
35
 
36
- ```from transformers import AutoModelForSequenceClassification, TFAutoModelForSequenceClassification
37
- from transformers import AutoTokenizer
38
- import numpy as np
39
- from scipy.special import expit
40
 
41
 
42
- MODEL = f"PavanDeepak/Topic_Classification"
43
- tokenizer = AutoTokenizer.from_pretrained(MODEL)
44
 
45
- model = AutoModelForSequenceClassification.from_pretrained(MODEL)
46
- class_mapping = model.config.id2label
47
 
48
- text = "I love chicken manchuria"
49
- tokens = tokenizer(text, return_tensors='pt')
50
- output = model(**tokens)
51
 
52
- scores = output[0][0].detach().numpy()
53
- scores = expit(scores)
54
- predictions = (scores >= 0.5) * 1
55
 
56
 
57
- for i in range(len(predictions)):
58
- if predictions[i]:
59
- print(class_mapping[i])```
 
33
 
34
  ## Example
35
 
36
+ ```from transformers import AutoModelForSequenceClassification, TFAutoModelForSequenceClassification```
37
+ ```from transformers import AutoTokenizer
38
+ ```import numpy as np
39
+ ```from scipy.special import expit
40
 
41
 
42
+ ```MODEL = f"PavanDeepak/Topic_Classification"
43
+ ```tokenizer = AutoTokenizer.from_pretrained(MODEL)
44
 
45
+ ```model = AutoModelForSequenceClassification.from_pretrained(MODEL)
46
+ ```class_mapping = model.config.id2label
47
 
48
+ ```text = "I love chicken manchuria"
49
+ ```tokens = tokenizer(text, return_tensors='pt')
50
+ ```output = model(**tokens)
51
 
52
+ ```scores = output[0][0].detach().numpy()
53
+ ```scores = expit(scores)
54
+ ```predictions = (scores >= 0.5) * 1
55
 
56
 
57
+ ```for i in range(len(predictions)):
58
+ ```if predictions[i]:
59
+ ```print(class_mapping[i])```