Update rembg module and docker file

2023-06-20 03:01:42 -07:00
parent 3d77324ad4
commit dff51fd9ac
2 changed files with 10 additions and 4 deletions
@@ -45,4 +45,7 @@ COPY ./agent_jarvis /root/jarvis
 COPY ./example_modules /root/example_modules
 COPY ./example_services /root/example_services

+# Download the u2net model
+RUN cd /root/example_modules/demo_modules/ && python rembg.module.py
+
 CMD ["bash", "/root/jarvis/supervisor.d/entrypoint.sh"]