diff --git a/compose.yaml b/compose.yaml
index 84a4eb0..441f206 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -15,7 +15,8 @@ services:
       - NVIDIA_DRIVER_CAPABILITIES=all
       - NVIDIA_VISIBLE_DEVICES=all
     restart: always
-    network_mode: "container:wireguard"  
+    ports:
+      - 8096:8096
     devices:
       - /dev/nvidia0:/dev/nvidia0
       - /dev/nvidiactl:/dev/nvidiactl