etadevosyan commited on
Commit
862f904
·
1 Parent(s): f003731

Json format

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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.components.Textbox(type='json',label='Дополнительные параметры услуги')],
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
  ['Прием (осмотр, консультация) - врача -оториноларинголога Первичный, рекомендации'],