app.py

import streamlit as st
from transformers import pipeline
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)
to_speech = TextToSpeech("NeuML/ljspeech-jets-onnx")
# Streamlit application title
st.title("Reading Ham")
st.write("Classification for Spam Email: spam or ham?")
# Text input for user to enter the text to classify
text = st.text_area("Paste the email to classify", "")
# Perform text classification when the user clicks the "Classify" button
if st.button("Classify"):
    # Perform text classification on the input text
    result = classifier(text)[0]
    # Display the classification result
    spam = "LABEL_1"
    ham = "LABEL_0"
    if result['label'] == spam:
        #st.write("Text:", text)
        st.write("Spam!")
    else:
        st.write("Ham!")
        speech = to_speech(text)
        st.audio(speech, sample_rate=16000)