aemin commited on
Commit
c4d15fc
1 Parent(s): c7696c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -87,8 +87,13 @@ elif selected_model== "ner_conll_elmo":
87
  st.markdown("<h2>"+app_description+"</h2>" , unsafe_allow_html=True)
88
  st.markdown("**`PER`** **,** **`LOC`** **,** **`ORG`** **,** **`MISC` **", unsafe_allow_html=True)
89
 
90
-
91
-
 
 
 
 
 
92
 
93
 
94
  elif selected_model=="ner_conll_albert_large_uncased":
 
87
  st.markdown("<h2>"+app_description+"</h2>" , unsafe_allow_html=True)
88
  st.markdown("**`PER`** **,** **`LOC`** **,** **`ORG`** **,** **`MISC` **", unsafe_allow_html=True)
89
 
90
+ elif selected_model== "ner_mit_movie_complex_distilbert_base_cased":
91
+ app_title= "Detect up to 12 entity types in movie domain texts"
92
+ app_description= "Named Entity Recognition model aimed to detect up to 12 entity types from movie domain texts. This model was trained on the MIT Movie Corpus complex queries dataset to detect movie trivia using Spark NLP, and it is available in Spark NLP Models hub (https://nlp.johnsnowlabs.com/models)"
93
+ st.title(app_title)
94
+ st.markdown("<h2>"+app_description+"</h2>" , unsafe_allow_html=True)
95
+ st.markdown("""**`ACTOR`** **,** **`AWARD`** **,** **`CHARACTER_NAME`** **,** **`DIRECTOR`** **,** **`GENRE`** **,** **`OPINION`** **,** **`ORIGIN`** **,** **`PLOT`**,
96
+ **`QUOTE`** **,** **`RELATIONSHIP`** **,** **`SOUNDTRACK`** **,** **`YEAR` **""", unsafe_allow_html=True)
97
 
98
 
99
  elif selected_model=="ner_conll_albert_large_uncased":