File size: 1,294 Bytes
d187b57 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
#!/bin/bash
# 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 "$@" |