Update app.py

app.py

@@ -4,7 +4,7 @@ import soundfile as sf
 from txtai.pipeline import TextToSpeech
 
 # Load the text classification model pipeline, filter out the spam and leave the ham
-classifier = pipeline("text-classification", model='JustHuggingFaces/OptimalSpamDetect', return_all_scores=True)
+classifier = pipeline("text-classification", model='JustHuggingFaces/OptimalSpamDetect')
 to_speech = TextToSpeech("NeuML/ljspeech-jets-onnx")
 # Streamlit application title
 st.title("Reading Ham")
@@ -21,7 +21,9 @@ if st.button("Classify"):
     if result['label'] == spam:
         #st.write("Text:", text)
         st.write("Spam!")
+        #st.write("Label: ", result['label'])
     else:
         st.write("Ham!")
+        #st.write("Label: ", result['label'])
         speech = to_speech(text)
         st.audio(speech, sample_rate=16000)