FROM tuvbui/torchcpu:torch111
ADD cldm ./cldm
ADD flae ./flae
ADD ldm ./ldm
ADD tools ./tools
ADD pages ./pages
ADD Embed_Secret.py .

EXPOSE 7860
CMD streamlit run Embed_Secret.py --server.enableXsrfProtection=false --server.port 7860 -- --weight https://kahlan.cvssp.org/data/Flickr25K/tubui/stega/unet160b/epoch=000099-step=000309999.ckpt --config https://kahlan.cvssp.org/data/Flickr25K/tubui/stega/unet160b/-project.yaml