Update app.py

app.py

@@ -71,14 +71,14 @@ def setModel(model_checkpoint):
 Run_Button = st.button("Run", key=None)
 if Run_Button == True:
     
-    zstc_pipeline = setModel(model_checkpoint)
-    output = zstc_pipeline(sequences=selected_text, candidate_labels=selected_labels)
-    output_labels = output["labels"]
-    output_scores = output["scores"]
-
-    st.header("Result")
-    import plotly.graph_objects as go
-    fig = go.Figure([go.Bar(x=output_labels, y=output_scores)])
-    st.plotly_chart(fig, use_container_width=False, sharing="streamlit")
-
+    with st.spinner('Model is running...'):
+        zstc_pipeline = setModel(model_checkpoint)
+        output = zstc_pipeline(sequences=selected_text, candidate_labels=selected_labels)
+        output_labels = output["labels"]
+        output_scores = output["scores"]
+
+        st.header("Result")
+        fig = go.Figure([go.Bar(x=output_labels, y=output_scores)])
+        st.plotly_chart(fig, use_container_width=False, sharing="streamlit")
+        st.success('Done!')