Spaces:
Runtime error
Runtime error
| # Test script for inference.py | |
| # Run this locally before submitting to catch errors early | |
| set -e | |
| echo "=== Testing inference.py locally ===" | |
| # Check required environment variables | |
| if [ -z "$HF_TOKEN" ] && [ -z "$API_KEY" ]; then | |
| echo "ERROR: HF_TOKEN or API_KEY environment variable is required" | |
| exit 1 | |
| fi | |
| # Set defaults for testing | |
| export API_BASE_URL="${API_BASE_URL:-https://router.huggingface.co/v1}" | |
| export MODEL_NAME="${MODEL_NAME:-openai/gpt-4o-mini}" | |
| # Option 1: Test with Docker image (if available) | |
| if [ -n "$IMAGE_NAME" ]; then | |
| echo "Testing with Docker image: $IMAGE_NAME" | |
| python inference.py | |
| exit $? | |
| fi | |
| # Option 2: Test with running server | |
| if [ -n "$SPACE_URL" ]; then | |
| echo "Testing with server: $SPACE_URL" | |
| python inference.py | |
| exit $? | |
| fi | |
| # Option 3: Start local server first | |
| echo "No IMAGE_NAME or SPACE_URL set." | |
| echo "Please either:" | |
| echo " 1. Set IMAGE_NAME to your Docker image name" | |
| echo " 2. Set SPACE_URL to your running server URL" | |
| echo " 3. Start the server locally first with: python -m server.app" | |
| exit 1 | |