Spaces:
Runtime error
Runtime error
AmitGarage
commited on
Commit
•
70b485b
1
Parent(s):
d2ef806
Update app.py
Browse files
app.py
CHANGED
@@ -18,13 +18,12 @@ def main(models: str = None, default_text: str = None):
|
|
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"
|
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 Price'):
|
27 |
-
# st.success(f'The predicted price of the diamond is USD')
|
28 |
spacy_streamlit.visualize(models, default_text, visualizers=["ner"], ner_labels=labels)
|
29 |
|
30 |
|
|
|
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 Price'):
|
|
|
27 |
spacy_streamlit.visualize(models, default_text, visualizers=["ner"], ner_labels=labels)
|
28 |
|
29 |
|