Update pages/2 Topic Modeling.py

pages/2 Topic Modeling.py

@@ -174,11 +174,11 @@ if uploaded_file is not None:
                  coherence_lda = coherence_model_lda.get_coherence()
                  vis = pyLDAvis.gensim_models.prepare(lda_model, corpus, id2word)
                  py_lda_vis_html = pyLDAvis.prepared_data_to_html(vis)
-                 return py_lda_vis_html, coherence_lda
+                 return py_lda_vis_html, coherence_lda, vis
                    
               with st.spinner('Performing computations. Please wait ...'):
                    try:
-                        py_lda_vis_html, coherence_lda = pylda(extype)
+                        py_lda_vis_html, coherence_lda, vis = pylda(extype)
                         st.write('Coherence: ', (coherence_lda))
                         st.components.v1.html(py_lda_vis_html, width=1500, height=800)
                         st.markdown('Copyright (c) 2015, Ben Mabey. https://github.com/bmabey/pyLDAvis')
@@ -202,7 +202,7 @@ if uploaded_file is not None:
                                   )
                         img_lda(vis)    
                        
-                   except ValueError:
+                   except NameError:
                         st.warning('🖱️ Please click Submit')
 
          with tab2: