predict = intent.predict_class(query) intent_name = intent.labels[predict] # 개체명 인식 from models.ner.NerModel import NerModel ner = NerModel(model_name='models/ner/ner_model.h5', proprocess=p) predicts = ner.predict(query) ner_tags = ner.predict_tags(query) print("질문 : ", query) print("=" * 100) print("의도 파악 : ", intent_name) print("개체명 인식 : ", predicts) print("답변 검색에 필요한 NER 태그 : ", ner_tags) print("=" * 100) # 답변 검색 from utils.FindAnswer import FindAnswer try: f = FindAnswer(db) answer_text, answer_image = f.search(intent_name, ner_tags) answer = f.tag_to_word(predicts, answer_text) except: answer = "죄송해요 무슨 말인지 모르겠어요" print("답변 : ", answer) db.close() # 디비 연결 끊음
preprocess=p) predict = intent.predict_class(query) intent_name = intent.labels[predict] # 개체명 인식 from models.ner.NerModel import NerModel ner = NerModel(model_name='../models/ner/ner_model.h5', preprocess=p) predicts = ner.predict(query) ner_tags = ner.predict_tags(query) print("질문 : ", query) print("=" * 40) print("의도 파악 : ", intent_name) print("개체명 인식 : ", predicts) print("답변 검색에 필요한 NER 태그", ner_tags) print("=" * 40) # 답변 검색 from utils.FindAnswer import FindAnswer try: f = FindAnswer(db) answer_text, answer_image = f.search(intent_name, ner_tags) answer = f.tag_to_word(predicts, answer_text) except: answer = "죄송해요, 무슨 말인지 모르겠어요." print("답변 : ", answer) db.close() # DB 연결 끊음