# Streamlit Launcher Script for Toxic Comment Classifier | |
# This script launches the Streamlit version of the application | |
echo "π Starting Toxic Comment Classifier - Streamlit Edition" | |
echo "π Loading model and dependencies..." | |
# Check for Python and Streamlit | |
if ! command -v python3 &> /dev/null; then | |
echo "β Python 3 is not installed. Please install Python 3 to run this application." | |
exit 1 | |
fi | |
if ! python3 -c "import streamlit" &> /dev/null; then | |
echo "β οΈ Streamlit not found. Attempting to install dependencies..." | |
pip install -r requirements.txt | |
fi | |
# Set default environment variables if not already set | |
export ONNX_MODEL_PATH=${ONNX_MODEL_PATH:-"weights/toxic_classifier.onnx"} | |
export PYTORCH_MODEL_DIR=${PYTORCH_MODEL_DIR:-"weights/toxic_classifier_xlm-roberta-large"} | |
# Set Streamlit environment variables to reduce errors | |
export STREAMLIT_SERVER_WATCH_ONLY_USER_CONTENT=true | |
export STREAMLIT_SERVER_HEADLESS=true | |
# Suppress TensorFlow warnings | |
export TF_CPP_MIN_LOG_LEVEL=2 | |
export TF_ENABLE_ONEDNN_OPTS=0 | |
# Run the Streamlit app with disabled hot-reload to avoid PyTorch class errors | |
echo "β Launching Streamlit application..." | |
streamlit run streamlit_app.py --server.port=8501 --server.address=0.0.0.0 --server.runOnSave=false "$@" |