{inputs, ...}: {
  flake.nixosModules.ollama = {
    pkgs,
    lib,
    ...
  }: {
    services.ollama = {
      enable = true;
      package = pkgs.ollama-rocm;
      loadModels = ["gemma4:31b"];
    };

    services.open-webui.enable = true;
  };
}