Commit
·
862f904
1
Parent(s):
f003731
Json format
Browse files
app.py
CHANGED
@@ -5,6 +5,7 @@ import pickle
|
|
5 |
import pandas as pd
|
6 |
from service_dops_api.dops_config import ServiceDopsConfig
|
7 |
from service_dops_api.dops_classifier import DopsClassifier
|
|
|
8 |
HF_TOKEN = os.getenv('HF_TOKEN')
|
9 |
tokenizer_cat = BertTokenizer.from_pretrained("warleagle/service_name_categorizer",
|
10 |
token=HF_TOKEN)
|
@@ -40,11 +41,12 @@ def service_pipeline(input_text):
|
|
40 |
else:
|
41 |
doctor_spec_result = doctor_spec_predict(input_text)
|
42 |
dops_result = dops_predict(input_text)
|
|
|
43 |
return categoriser_result,doctor_spec_result,dops_result
|
44 |
demo = gr.Interface(fn=service_pipeline,inputs=gr.components.Textbox(label='Название услуги'),
|
45 |
outputs=[gr.components.Textbox(label='Относится ли данная услуга к приёму специалиста'),
|
46 |
gr.components.Textbox(label='Специальность врача'),
|
47 |
-
gr.
|
48 |
examples=[
|
49 |
['Врач-офтальмолог (высшая категория/кандидат медицинских наук), первичный приём'],
|
50 |
['Прием (осмотр, консультация) - врача -оториноларинголога Первичный, рекомендации'],
|
|
|
5 |
import pandas as pd
|
6 |
from service_dops_api.dops_config import ServiceDopsConfig
|
7 |
from service_dops_api.dops_classifier import DopsClassifier
|
8 |
+
import json
|
9 |
HF_TOKEN = os.getenv('HF_TOKEN')
|
10 |
tokenizer_cat = BertTokenizer.from_pretrained("warleagle/service_name_categorizer",
|
11 |
token=HF_TOKEN)
|
|
|
41 |
else:
|
42 |
doctor_spec_result = doctor_spec_predict(input_text)
|
43 |
dops_result = dops_predict(input_text)
|
44 |
+
dops_result = json.dumps(dops_result,indent=4)
|
45 |
return categoriser_result,doctor_spec_result,dops_result
|
46 |
demo = gr.Interface(fn=service_pipeline,inputs=gr.components.Textbox(label='Название услуги'),
|
47 |
outputs=[gr.components.Textbox(label='Относится ли данная услуга к приёму специалиста'),
|
48 |
gr.components.Textbox(label='Специальность врача'),
|
49 |
+
gr.Textbox(label='Дополнительные параметры услуги')],
|
50 |
examples=[
|
51 |
['Врач-офтальмолог (высшая категория/кандидат медицинских наук), первичный приём'],
|
52 |
['Прием (осмотр, консультация) - врача -оториноларинголога Первичный, рекомендации'],
|