Spaces:
Runtime error
Runtime error
Commit
•
7222e0d
1
Parent(s):
70b485b
Update app.py
Browse files
app.py
CHANGED
@@ -7,23 +7,13 @@ from scripts.torch_ner_pipe import make_torch_entity_recognizer
|
|
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 |
-
#carat = st.number_input('Carat Weight:', min_value=0.1, max_value=10.0, value=1.0)
|
12 |
-
#cut = st.selectbox('Cut Rating:', ['Fair', 'Good', 'Very Good', 'Premium', 'Ideal'])
|
13 |
-
#color = st.selectbox('Color Rating:', ['J', 'I', 'H', 'G', 'F', 'E', 'D'])
|
14 |
-
#clarity = st.selectbox('Clarity Rating:', ['I1', 'SI2', 'SI1', 'VS2', 'VS1', 'VVS2', 'VVS1', 'IF'])
|
15 |
-
#depth = st.number_input('Diamond Depth Percentage:', min_value=0.1, max_value=100.0, value=1.0)
|
16 |
-
#table = st.number_input('Diamond Table Percentage:', min_value=0.1, max_value=100.0, value=1.0)
|
17 |
-
#x = st.number_input('Diamond Length (X) in mm:', min_value=0.1, max_value=100.0, value=1.0)
|
18 |
-
#y = st.number_input('Diamond Width (Y) in mm:', min_value=0.1, max_value=100.0, value=1.0)
|
19 |
-
#z = st.number_input('Diamond Height (Z) in mm:', min_value=0.1, max_value=100.0, value=1.0)
|
20 |
|
21 |
models = ["training/model-best","training_trf/model-best"]
|
22 |
default_text = "The patient had surgery."
|
23 |
models = [name.strip() for name in models.split(",")]
|
24 |
labels = ["person", "problem", "pronoun", "test", "treatment"]
|
25 |
|
26 |
-
#if st.button('Predict
|
27 |
spacy_streamlit.visualize(models, default_text, visualizers=["ner"], ner_labels=labels)
|
28 |
|
29 |
|
|
|
7 |
|
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"]
|
15 |
|
16 |
+
#if st.button('Predict entities'):
|
17 |
spacy_streamlit.visualize(models, default_text, visualizers=["ner"], ner_labels=labels)
|
18 |
|
19 |
|