Harsh502s commited on
Commit
0797b32
·
1 Parent(s): f24359a

Add tag assignment spinners to supervised and

Browse files
Files changed (1) hide show
  1. Pages/Models.py +9 -5
Pages/Models.py CHANGED
@@ -79,8 +79,9 @@ def supervised_page():
79
  text = st.text_area("Enter text to assign tags", height=200, key="supervised_text")
80
  text = clean_text(text)
81
  if st.button("Assign tags", key="supervised_button"):
82
- tags = tag_cnn_model(text)[0]
83
- tagger_component("Tags:", tags)
 
84
 
85
 
86
  # Display the unsupervised model using bertopic page of the app
@@ -94,7 +95,8 @@ def unsupervised_page_bertopic():
94
  "Enter number of tags to assign", value=5, key="unsupervised_n_bertopic"
95
  )
96
  if st.button("Assign tags", key="unsupervised_button_bertopic"):
97
- output_bertopic(text, n)
 
98
 
99
 
100
  def unsupervised_page_keybert():
@@ -107,7 +109,8 @@ def unsupervised_page_keybert():
107
  "Enter number of tags to assign", value=5, key="unsupervised_n_keybert"
108
  )
109
  if st.button("Assign tags", key="unsupervised_button_keybert"):
110
- output_keybert(text, n)
 
111
 
112
 
113
  # Display the unsupervised model using bertopic page of the app
@@ -121,7 +124,8 @@ def unsupervised_page_mmr():
121
  "Enter number of tags to assign", value=5, key="unsupervised_n_mmr"
122
  )
123
  if st.button("Assign tags", key="unsupervised_button_mmr"):
124
- output_mmr(text, n)
 
125
 
126
 
127
  # Display the model page of the app
 
79
  text = st.text_area("Enter text to assign tags", height=200, key="supervised_text")
80
  text = clean_text(text)
81
  if st.button("Assign tags", key="supervised_button"):
82
+ with st.spinner("Assigning tags..."):
83
+ tags = tag_cnn_model(text)[0]
84
+ tagger_component("Tags:", tags)
85
 
86
 
87
  # Display the unsupervised model using bertopic page of the app
 
95
  "Enter number of tags to assign", value=5, key="unsupervised_n_bertopic"
96
  )
97
  if st.button("Assign tags", key="unsupervised_button_bertopic"):
98
+ with st.spinner("Assigning tags..."):
99
+ output_bertopic(text, n)
100
 
101
 
102
  def unsupervised_page_keybert():
 
109
  "Enter number of tags to assign", value=5, key="unsupervised_n_keybert"
110
  )
111
  if st.button("Assign tags", key="unsupervised_button_keybert"):
112
+ with st.spinner("Assigning tags..."):
113
+ output_keybert(text, n)
114
 
115
 
116
  # Display the unsupervised model using bertopic page of the app
 
124
  "Enter number of tags to assign", value=5, key="unsupervised_n_mmr"
125
  )
126
  if st.button("Assign tags", key="unsupervised_button_mmr"):
127
+ with st.spinner("Assigning tags..."):
128
+ output_mmr(text, n)
129
 
130
 
131
  # Display the model page of the app