🚀 Changed model to use for the U2NETP

AlbertSuarez · Dec 8, 2020 · 61f0c06 · 61f0c06
1 parent f5410fe
commit 61f0c06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -44,7 +44,7 @@ services:
       - object_cut
     restart: always
     environment:
-      - MODEL=BASNet  # Can also be `U2NET`
+      - MODEL=U2NETP  # Can also be `BASNet`, `U2NET` or `U2NETPORTRAIT`
     labels:
       - 'traefik.enable=true'
       - 'traefik.docker.network=object_cut'