You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Увидел ссылку на модель, но карточка пуста, а в имплементации только инит с AutoModel. Подскажите, как запустить на предикт модель на своих данных?
Пытался так:
tokenizer = AutoTokenizer.from_pretrained("cimm-kzn/rudr-bert")
model = AutoModel.from_pretrained("cimm-kzn/rudr-bert")
example = "Мне прописали Ингверин как имуномодулятор."
Но это не работает.
Вручную получаю выход с модели, но дальше не понимаю, что делать:
with torch.no_grad():
out = model(**tokenizer(example, return_tensors='pt').to(model.device))
The text was updated successfully, but these errors were encountered:
In Examples (see readme), you can find a Colab notebook on how to use NER models for the detection of named entities such as drugs and adverse drug reactions. Hugging Face models are language models, so need to train on your target data additionally.
Увидел ссылку на модель, но карточка пуста, а в имплементации только инит с AutoModel. Подскажите, как запустить на предикт модель на своих данных?
Пытался так:
tokenizer = AutoTokenizer.from_pretrained("cimm-kzn/rudr-bert")
model = AutoModel.from_pretrained("cimm-kzn/rudr-bert")
example = "Мне прописали Ингверин как имуномодулятор."
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
ner_results = nlp(example)
print(ner_results)
Но это не работает.
Вручную получаю выход с модели, но дальше не понимаю, что делать:
with torch.no_grad():
out = model(**tokenizer(example, return_tensors='pt').to(model.device))
The text was updated successfully, but these errors were encountered: