system HF staff commited on
Commit
b89dc8f
1 Parent(s): 37df727

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -11,7 +11,9 @@ tags:
11
  - text-classification
12
  - industry tags
13
  - buisiness description
14
- liscence: "mit"
 
 
15
  ---
16
 
17
  # distilbert-business-domain-classification
@@ -26,8 +28,8 @@ Trained on 7000 samples of Business Descriptions and associated labels of compan
26
  ```python
27
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
28
 
29
- tokenizer = AutoTokenizer.from_pretrained("sampathkethineedi/distilbert-business-domain-classification")
30
- model = AutoModelForSequenceClassification.from_pretrained("distilbert-business-domain-classification")
31
 
32
  industry_tags = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
33
 
 
11
  - text-classification
12
  - industry tags
13
  - buisiness description
14
+ - multi label classification
15
+ liscence: "mit"
16
+ inference: false
17
  ---
18
 
19
  # distilbert-business-domain-classification
 
28
  ```python
29
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
30
 
31
+ tokenizer = AutoTokenizer.from_pretrained("sampathkethineedi/industry-classification")
32
+ model = AutoModelForSequenceClassification.from_pretrained("industry-classification")
33
 
34
  industry_tags = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
35