etadevosyan commited on
Commit
527b3f6
·
1 Parent(s): 64aec53

Edited output

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. service_dops_api/dops_classifier.py +9 -6
app.py CHANGED
@@ -13,6 +13,7 @@ model_cat = BertForSequenceClassification.from_pretrained('warleagle/service_nam
13
  tokenizer_spec = BertTokenizer.from_pretrained("warleagle/specialists_categorizer_model",
14
  token=HF_TOKEN)
15
  model_spec = BertForSequenceClassification.from_pretrained('warleagle/specialists_categorizer_model',token=HF_TOKEN)
 
16
  def categoriser_predict(input_text):
17
  clf = pipeline("text-classification", model=model_cat, tokenizer=tokenizer_cat)
18
  predictions = clf(input_text)
 
13
  tokenizer_spec = BertTokenizer.from_pretrained("warleagle/specialists_categorizer_model",
14
  token=HF_TOKEN)
15
  model_spec = BertForSequenceClassification.from_pretrained('warleagle/specialists_categorizer_model',token=HF_TOKEN)
16
+
17
  def categoriser_predict(input_text):
18
  clf = pipeline("text-classification", model=model_cat, tokenizer=tokenizer_cat)
19
  predictions = clf(input_text)
service_dops_api/dops_classifier.py CHANGED
@@ -15,13 +15,16 @@ class DopsClassifier:
15
  result[self.config.dops_default_values[dop_name]] = 0
16
  return result
17
 
18
- def convert_search_to_human(self,dict_from_search):
19
- return [key for key, value in dict_from_search.items() if value == 1]
20
-
 
 
 
21
  def run_all_dops(self,text):
22
  all_dops = self.config.option_patterns_dict.keys()
23
- result_dict = {}
24
  for dop in all_dops:
25
  temp_dop = self.run_regular_search(text,dop)
26
- result_dict[dop] = self.convert_search_to_human(temp_dop)
27
- return result_dict
 
15
  result[self.config.dops_default_values[dop_name]] = 0
16
  return result
17
 
18
+ def convert_search_to_human(self,dict_from_search,dop_name):
19
+ result = {}
20
+ result['dop_name'] = dop_name
21
+ result['values'] = [key for key, value in dict_from_search.items() if value == 1]
22
+ result['default_values'] = list(dict_from_search.keys())
23
+ return result
24
  def run_all_dops(self,text):
25
  all_dops = self.config.option_patterns_dict.keys()
26
+ result_list = []
27
  for dop in all_dops:
28
  temp_dop = self.run_regular_search(text,dop)
29
+ result_list.append(self.convert_search_to_human(temp_dop,dop))
30
+ return result_list