Spaces:
Runtime error
Runtime error
Commit
•
246ed34
1
Parent(s):
ba9e424
Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,8 @@ from scripts.torch_ner_pipe import make_torch_entity_recognizer
|
|
6 |
|
7 |
|
8 |
def main(models: str = None, default_text: str = None):
|
|
|
|
|
9 |
models = "training_trf/model-best"
|
10 |
test = "The patient had surgery."
|
11 |
models = [name.strip() for name in models.split(",")]
|
@@ -13,8 +15,6 @@ def main(models: str = None, default_text: str = None):
|
|
13 |
spacy_streamlit.visualize(
|
14 |
models, default_text, visualizers=["ner"], ner_labels=labels
|
15 |
)
|
16 |
-
st.title('NER Predictor')
|
17 |
-
st.header('Enter the characteristics of the diamond:')
|
18 |
|
19 |
|
20 |
if __name__ == "__main__":
|
|
|
6 |
|
7 |
|
8 |
def main(models: str = None, default_text: str = None):
|
9 |
+
st.title('NER Predictor')
|
10 |
+
st.header('Enter the characteristics of the diamond:')
|
11 |
models = "training_trf/model-best"
|
12 |
test = "The patient had surgery."
|
13 |
models = [name.strip() for name in models.split(",")]
|
|
|
15 |
spacy_streamlit.visualize(
|
16 |
models, default_text, visualizers=["ner"], ner_labels=labels
|
17 |
)
|
|
|
|
|
18 |
|
19 |
|
20 |
if __name__ == "__main__":
|