diff --git a/backend/open_webui/__init__.py b/backend/open_webui/__init__.py
index 743656d9e..167f0fb60 100644
--- a/backend/open_webui/__init__.py
+++ b/backend/open_webui/__init__.py
@@ -41,6 +41,7 @@ def serve(
         )
         try:
             import torch
+
             assert torch.cuda.is_available(), "CUDA not available"
             typer.echo("CUDA seems to be working")
         except Exception as e:
diff --git a/backend/open_webui/env.py b/backend/open_webui/env.py
index df5597cbf..cf45e725c 100644
--- a/backend/open_webui/env.py
+++ b/backend/open_webui/env.py
@@ -38,6 +38,7 @@ USE_CUDA = os.environ.get("USE_CUDA_DOCKER", "false")
 if USE_CUDA.lower() == "true":
     try:
         import torch
+
         assert torch.cuda.is_available(), "CUDA not available"
         DEVICE_TYPE = "cuda"
     except Exception as e: