change the data type of top_labels

app.py

@@ -98,7 +98,7 @@ def classify(text):
     return results
 
 # Streamlit app
-st.title("Text Classification Demo")
+st.title("Text Classification.")
 st.write("Enter some text, and the model will classify it.")
 
 text_input = st.text_input("Text Input")
@@ -106,5 +106,5 @@ if st.button("Classify"):
     predictions = classify(text_input)
     for prediction in predictions:
        # st.write(f"Segment Text: {prediction['segment_text']}")
-        st.write(f"Label: {prediction['top_labels']}")
+        st.write(f"Label: {list(prediction['top_labels'])}")
         st.write(f"Probability: {prediction['top_probabilities']}")