stratagem / server /app.py
Astro-Dude's picture
Upload server/app.py with huggingface_hub
9e72303 verified
Raw
History Blame Contribute Delete
604 Bytes
"""
Stratagem: Cybersecurity Incident Response — Server Entry Point
"""
import sys
import os
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from openenv.core.env_server import create_app
from models import IncidentAction, IncidentObservation
from environment import StratagemEnvironment
app = create_app(
env=StratagemEnvironment,
action_cls=IncidentAction,
observation_cls=IncidentObservation,
env_name="stratagem",
)
def main() -> None:
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860)
if __name__ == "__main__":
main()