Spaces:
Runtime error
Runtime error
AmitGarage
commited on
Commit
•
28bf1a8
1
Parent(s):
7222e0d
Update app.py
Browse files
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 =
|
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"]
|