AmitGarage commited on
Commit
28bf1a8
1 Parent(s): 7222e0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from scripts.torch_ner_pipe import make_torch_entity_recognizer
8
  def main(models: str = None, default_text: str = None):
9
  st.title('NER Predictor')
10
 
11
- models = ["training/model-best","training_trf/model-best"]
12
  default_text = "The patient had surgery."
13
  models = [name.strip() for name in models.split(",")]
14
  labels = ["person", "problem", "pronoun", "test", "treatment"]
 
8
  def main(models: str = None, default_text: str = None):
9
  st.title('NER Predictor')
10
 
11
+ models = "training/model-best,training_trf/model-best"
12
  default_text = "The patient had surgery."
13
  models = [name.strip() for name in models.split(",")]
14
  labels = ["person", "problem", "pronoun", "test", "treatment"]