Эта модель предназначена для классификации диалоговых вопросов на QA и Dialog. Модель может применятся в различных чат-ботах для управления режимом диалога и интернет-поиска.
```python
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification

use_cuda = torch.cuda.is_available()
device = torch.device("cuda" if use_cuda else "cpu")
tokenizer = AutoTokenizer.from_pretrained('Den4ikAI/ruBert_tiny_qa_detection')
model = AutoModelForSequenceClassification.from_pretrained('Den4ikAI/ruBert_tiny_qa_detection')
model.to(device)
model.eval()


def classify(text):
    inputs = tokenizer(text, max_length=128, add_special_tokens=False, return_tensors='pt').to(device)
    with torch.no_grad():
        logits = model(**inputs).logits
        probas = torch.sigmoid(logits)[0].cpu().detach().numpy()
    return probas
print(classify(input(':> ')))
```

{"language": ["ru"], "license": "mit", "widget": [{"text": "\u041f\u043e\u0447\u0435\u043c\u0443 \u0442\u0440\u0430\u0432\u0430 \u0437\u0435\u043b\u0435\u043d\u0430\u044f?"}, {"text": "\u041a\u0430\u043a \u0434\u0435\u043b\u0430?"}, {"text": "\u041a\u043e\u0433\u0434\u0430 \u0440\u043e\u0434\u0438\u043b\u0441\u044f \u041f\u0443\u0448\u043a\u0438\u043d?"}, {"text": "\u041a\u0430\u043a \u0442\u0435\u0431\u044f \u0437\u043e\u0432\u0443\u0442?"}]}
transformers pytorch safetensors bert text-classification ru license:mit autotrain_compatible endpoints_compatible region:us
TAGS
#transformers #pytorch #safetensors #bert #text-classification #ru #license-mit #autotrain_compatible #endpoints_compatible #region-us 

Эта модель предназначена для классификации диалоговых вопросов на QA и Dialog. Модель может применятся в различных чат-ботах для управления режимом диалога и интернет-поиска.


TAGS
#transformers #pytorch #safetensors #bert #text-classification #ru #license-mit #autotrain_compatible #endpoints_compatible #region-us 

TAGS
#transformers #pytorch #safetensors #bert #text-classification #ru #license-mit #autotrain_compatible #endpoints_compatible #region-us 
