pasukka
/

auto-filters-term-classifier-v.0.2

Text Classification

Inference Endpoints

Model card Files Files and versions Community

pasukka commited on Sep 2, 2024

Commit

73ded54

·

verified ·

1 Parent(s): 66313f9

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ base_model: ai-forever/ruRoberta-large
 ## Usage
-```
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 model = AutoModelForSequenceClassification.from_pretrained("pasukka/auto-filters-term-classifier-v.0.2")
@@ -36,7 +36,8 @@ tokenizer = AutoTokenizer.from_pretrained('ai-forever/ruRoberta-large')
 term = 'фильтр топливный'
 outputs = model.forward(**tokenizer(term, return_tensors='pt').to(device='cuda'))
 id = outputs.logits.argmax(dim=1)[0].item()
-model.config.id2label[id]
 ```
 Result:
 ```

 ## Usage
+```python
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 model = AutoModelForSequenceClassification.from_pretrained("pasukka/auto-filters-term-classifier-v.0.2")
 term = 'фильтр топливный'
 outputs = model.forward(**tokenizer(term, return_tensors='pt').to(device='cuda'))
 id = outputs.logits.argmax(dim=1)[0].item()
+print(model.config.id2label[id])
 ```
 Result:
 ```