document_redaction / entrypoint_router.py
seanpedrickcase's picture
Removed test event from entrypoint_router.py
7f2dc0f
raw
history blame
727 Bytes
import os
import subprocess
if __name__ == "__main__":
run_direct_mode = os.getenv("RUN_DIRECT_MODE", "0")
if run_direct_mode == "1":
# Invoke the lambda handler
from lambda_entrypoint import lambda_handler
else:
# Gradio App execution
from app import app, max_queue_size, max_file_size # Replace with actual import if needed
from tools.auth import authenticate_user
if os.getenv("COGNITO_AUTH", "0") == "1":
app.queue(max_size=max_queue_size).launch(show_error=True, auth=authenticate_user, max_file_size=max_file_size)
else:
app.queue(max_size=max_queue_size).launch(show_error=True, inbrowser=True, max_file_size=max_file_size)