File size: 942 Bytes
e5dfae7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import os
import subprocess
if __name__ == "__main__":
run_direct_mode = os.getenv("RUN_DIRECT_MODE", "0")
if run_direct_mode == "1":
# Lambda execution or CLI invocation (Direct Mode)
from lambda_entrypoint import lambda_handler
# Simulate the Lambda event and context for local testing
event = os.getenv("LAMBDA_TEST_EVENT", '{}')
context = None # Add mock context if needed
response = lambda_handler(eval(event), context)
print(response)
else:
# Gradio App execution
from app import app # Replace with actual import if needed
if os.getenv("COGNITO_AUTH", "0") == "1":
app.queue(max_size=app.max_queue_size).launch(show_error=True, auth=app.authenticate_user, max_file_size=app.max_file_size)
else:
app.queue(max_size=app.max_queue_size).launch(show_error=True, inbrowser=True, max_file_size=app.max_file_size) |