diff --git a/modules/nixos/server/chat.nix b/modules/nixos/server/chat.nix
index ba8a377..340ed7d 100644
--- a/modules/nixos/server/chat.nix
+++ b/modules/nixos/server/chat.nix
@@ -15,7 +15,7 @@ in with lib; {
   config = mkIf cfg.enable {
     services.ollama = {
       enable = true;
-      loadModels = [ "deepseek-r1:7b" ];
+      loadModels = [ "deepseek-r1:1.5b" "deepseek-r1:7b" "deepseek-r1:8b" ];
 
     };
 
@@ -29,7 +29,7 @@ in with lib; {
         SCARF_NO_ANALYTICS = "True";
         OLLAMA_API_BASE_URL = "http://127.0.0.1:11434";
         # Disable authentication
-        WEBUI_AUTH = "False";
+        # WEBUI_AUTH = "False";
       };
 
       inherit port;
diff --git a/modules/nixos/server/nginx.nix b/modules/nixos/server/nginx.nix
index 54faea0..7322c39 100644
--- a/modules/nixos/server/nginx.nix
+++ b/modules/nixos/server/nginx.nix
@@ -90,7 +90,12 @@ in with lib; {
           enableACME = true;
         };
         proxy = port:
-          base { "/".proxyPass = "http://${cfg.ip}:${toString port}/"; };
+          base {
+            "/" = {
+              proxyPass = "http://${cfg.ip}:${toString port}/";
+              proxyWebsockets = true;
+            };
+          };
       in mapAttrs' (name: port:
         nameValuePair ("${name}.${cfg.domain}")
         # (proxy port // { default = true; })) cfg.ports;