File size: 263 Bytes
dcd93f5 |
1 2 3 4 5 6 7 8 9 10 11 |
from app.travel_resolver.libs.nlp.ner.models import LSTM_NER, BiLSTM_NER, CamemBERT_NER
import tensorflow as tf
print(tf.__version__)
ner_model = LSTM_NER()
sentence = "Je voudrais voyager de Nice à Clermont Ferrand."
print(ner_model.get_entities(sentence))
|