prajwal967 commited on
Commit
c839604
1 Parent(s): 193cf48

added 99.7 recall

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -177,8 +177,8 @@ class App(object):
177
  @staticmethod
178
  def _get_threshold_map():
179
  return {
180
- 'obi/deid_bert_i2b2':{"99.5": 4.656325975101986e-06},
181
- 'obi/deid_roberta_i2b2':{"99.5": 2.4362972672812125e-05}
182
  }
183
 
184
  @staticmethod
@@ -236,7 +236,7 @@ def deid(text, model, threshold):
236
  deid_text_remove = list(app.get_deid_text_removed(tmp.name, tmp_1.name))[0]['deid_text']
237
  return [highlight_text for highlight_text in App._get_highlights(deid_text)], deid_text_remove
238
 
239
- recall_choices = ["No threshold", "99.5"]
240
  recall_radio_input = gr.inputs.Radio(recall_choices, type="value", default=None, label='RECALL THRESHOLD')
241
 
242
  model_choices = list(App._get_model_map().keys())
 
177
  @staticmethod
178
  def _get_threshold_map():
179
  return {
180
+ 'obi/deid_bert_i2b2':{"99.5": 4.656325975101986e-06, "99.7":1.8982457699258832e-06},
181
+ 'obi/deid_roberta_i2b2':{"99.5": 2.4362972672812125e-05, "99.7":2.396420546444644e-06}
182
  }
183
 
184
  @staticmethod
 
236
  deid_text_remove = list(app.get_deid_text_removed(tmp.name, tmp_1.name))[0]['deid_text']
237
  return [highlight_text for highlight_text in App._get_highlights(deid_text)], deid_text_remove
238
 
239
+ recall_choices = ["No threshold", "99.5", "99.7"]
240
  recall_radio_input = gr.inputs.Radio(recall_choices, type="value", default=None, label='RECALL THRESHOLD')
241
 
242
  model_choices = list(App._get_model_map().keys())