Update src/webApp.py

src/webApp.py

@@ -480,8 +480,10 @@ if st.button("Analyse Review"):
         with st.spinner("Computing word importance..."):
             word_importance = get_word_importance(review, tokenizer, bert_model, s_label)
 
-        if word_fig:
-            st.plotly_chart(word_fig, use_container_width=True)
+        if word_importance:
+            word_fig = word_importance_chart(word_importance, s_label)
+            if word_fig:
+                st.plotly_chart(word_fig, use_container_width=True)
 
             # Plain English explanation of top words
             st.markdown("#### What the words mean for this prediction")